diff --git a/library/ipaclient.py b/library/ipaclient.py
index 46b33f43..df3189d5 100644
--- a/library/ipaclient.py
+++ b/library/ipaclient.py
@@ -138,10 +138,11 @@ tbd
 
 import os
 from six.moves.configparser import RawConfigParser
-
 from ansible.module_utils.basic import AnsibleModule
-
-from ipalib.install.sysrestore import SYSRESTORE_STATEFILE
+try:
+    from ipalib.install.sysrestore import SYSRESTORE_STATEFILE
+except ImportError:
+    from ipapython.sysrestore import SYSRESTORE_STATEFILE
 from ipaplatform.paths import paths