diff --git a/user_tools/tests/spark_rapids_tools_ut/resources/cluster/databricks/test-azure-instances-catalog.json b/user_tools/tests/spark_rapids_tools_ut/resources/cluster/databricks/test-azure-instances-catalog.json
index e69de29bb..85080e8e8 100644
--- a/user_tools/tests/spark_rapids_tools_ut/resources/cluster/databricks/test-azure-instances-catalog.json
+++ b/user_tools/tests/spark_rapids_tools_ut/resources/cluster/databricks/test-azure-instances-catalog.json
@@ -0,0 +1,22 @@
+{
+  "Standard_NC4as_T4_v3": {
+    "VCpuInfo": {
+      "DefaultVCpus": 4
+    },
+    "MemoryInfo": {
+      "SizeInMiB": 0
+    },
+    "GpuInfo": {
+      "GPUs": [
+        {
+          "Name": "",
+          "Manufacturer": "",
+          "Count": 1,
+          "MemoryInfo": {
+            "SizeInMiB": 0
+          }
+        }
+      ]
+    }
+  }
+}
\ No newline at end of file
diff --git a/user_tools/tests/test_diagnostic.py b/user_tools/tests/test_diagnostic.py
index be9326306..d15e17615 100644
--- a/user_tools/tests/test_diagnostic.py
+++ b/user_tools/tests/test_diagnostic.py
@@ -22,7 +22,7 @@
 from cli_test_helpers import ArgvContext, EnvironContext  # pylint: disable=import-error
 
 from spark_rapids_pytools import wrapper
-from .pyrapids_unit import conftest
+from .spark_rapids_tools_ut import conftest
 from .mock_cluster import mock_live_cluster