Update __init__.py

databricks-academy · Oct 2, 2022 · b9832b8 · b9832b8
1 parent ab38798
commit b9832b8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/dbacademy_gems/dbgems/__init__.py b/src/dbacademy_gems/dbgems/__init__.py
@@ -250,8 +250,8 @@ def check_for_latest_version(module: str, curriculum_workspaces_only=True) -> bo
     try:
         if curriculum_workspaces_only is False or is_curriculum_workspace():
             # Don't do anything unless this is in one of the Curriculum Workspaces
-            current_version = lookup_current_module_version("dbacademy-gems")[1:]
-            versions = lookup_all_module_versions("dbacademy-gems")
+            current_version = lookup_current_module_version(module)[1:]
+            versions = lookup_all_module_versions(module)
 
             if current_version == versions[-1]:
                 return True  # They match, all done!