diff --git a/library/ipaclient.py b/library/ipaclient.py
index 46b33f4340b29a2dce59eda1cdf154705d503896..df3189d569035df697787fb5257b55e8d23f82a9 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