for filename in files:
         filename = os.path.basename(filename).replace(".py", "")
         o = importlib.import_module(
-            "moon_db.migrate_repo.versions.{}".format(filename))
+            "python_moondb.migrate_repo.versions.{}".format(filename))
         logger.info("Command is {}".format(command))
         if command in ("upgrade", "u", "up"):
             logger.info(
-                "upgrading moon_db.migrate_repo.versions.{}".format(filename))
+                "upgrading python_moondb.migrate_repo.versions.{}".format(filename))
             o.upgrade(engine)
         elif command in ("downgrade", "d", "down"):
             logger.info(
-                "downgrading moon_db.migrate_repo.versions.{}".format(
+                "downgrading python_moondb.migrate_repo.versions.{}".format(
                     filename))
             o.downgrade(engine)
         else:
 
         mock_components.register_components(m)
         mock_keystone.register_keystone(m)
 
-        from moon_db.db_manager import init_engine, main
+        from python_moondb.db_manager import init_engine, main
         engine = init_engine()
         main("upgrade", logging.getLogger("db_manager"), engine)
         yield m
 
 
 
 def get_policies():
-    from moon_db.core import PolicyManager
+    from python_moondb.core import PolicyManager
     return PolicyManager.get_policies("admin")
 
 
 def add_policies(value=None):
-    from moon_db.core import PolicyManager
+    from python_moondb.core import PolicyManager
     if not value:
         value = {
             "name": "test_policiy",
 
 
 def delete_policies(uuid=None, name=None):
-    from moon_db.core import PolicyManager
+    from python_moondb.core import PolicyManager
     if not uuid:
         for policy_id, policy_value in get_policies():
             if name == policy_value['name']: