diff --git a/blue_geo/__init__.py b/blue_geo/__init__.py
index b9c61b4d..23f4e0fb 100644
--- a/blue_geo/__init__.py
+++ b/blue_geo/__init__.py
@@ -4,7 +4,7 @@
 
 DESCRIPTION = f"{ICON} AI for precise geospatial data analysis and visualization."
 
-VERSION = "4.154.1"
+VERSION = "4.155.1"
 
 REPO_NAME = "blue-geo"
 
diff --git a/blue_geo/catalog/classes.py b/blue_geo/catalog/classes.py
index 46490516..8e5aa800 100644
--- a/blue_geo/catalog/classes.py
+++ b/blue_geo/catalog/classes.py
@@ -33,5 +33,5 @@ def get_datacube_classes(
     return [
         datacube_class
         for datacube_class in list_of_datacube_classes
-        if isinstance(datacube_class.catalog, catalog_class)
+        if datacube_class.catalog.__class__ == catalog_class
     ]
diff --git a/blue_geo/tests/test_catalog.py b/blue_geo/tests/test_catalog.py
index 5cb07f2c..a80aa1c2 100644
--- a/blue_geo/tests/test_catalog.py
+++ b/blue_geo/tests/test_catalog.py
@@ -22,7 +22,7 @@ def test_list_of_catalog_classes(catalog_class: Type[GenericCatalog]):
 
 
 @pytest.mark.parametrize(
-    ["datacube_class", "datacube_id"],
+    ["datacube_id", "datacube_class"],
     [
         [datacube_id, datacube_class]
         for datacube_id, datacube_class in assets.datacubes.items()
@@ -30,8 +30,8 @@ def test_list_of_catalog_classes(catalog_class: Type[GenericCatalog]):
     ],
 )
 def test_list_of_datacube_classes(
-    datacube_class: Type[GenericDatacube],
     datacube_id: str,
+    datacube_class: Type[GenericDatacube],
 ):
     object_name = unique_object()
 
@@ -83,6 +83,6 @@ def test_get_datacube_classes(catalog_class: Type[GenericCatalog]):
     datacube_class_list = get_datacube_classes(catalog_class)
 
     assert all(
-        datacube_class.catalog == catalog_class
+        datacube_class.catalog.__class__ == catalog_class
         for datacube_class in datacube_class_list
     )