1 files changed, 5 insertions, 5 deletions
diff --git a/update.py b/update.py
index 44fcc85..5a19989 100644
--- a/update.py
+++ b/update.py
@@ -62,11 +62,7 @@ import re
 import shutil
 import sys
 
-try:
-    from openstack.common import cfg
-except:
-    sys.stderr.write("Try running update.sh")
-    raise
+from oslo.config import cfg
 
 opts = [
     cfg.ListOpt('modules',
@@ -150,6 +146,10 @@ def _copy_file(path, dest, base):
              '\"openstack\.common',
              '\"' + base + '.openstack.common')
 
+    _replace(dest,
+             'possible_topdir, "oslo",$',
+             'possible_topdir, "' + base + '",')
+
 
 def _copy_pyfile(path, base, dest_dir):
     _copy_file(path, _dest_path(path, base, dest_dir), base)