diff --git a/.github/workflows/CI.yml b/.github/workflows/CI.yml
index a753f80..c15bdf4 100644
--- a/.github/workflows/CI.yml
+++ b/.github/workflows/CI.yml
@@ -102,11 +102,20 @@ jobs:
         run: uv python install ${{ matrix.py }} && uv python pin ${{ matrix.py }} && uv venv --python ${{ matrix.py }}
       - name: Override ray==${{ matrix.ray }} dagster==${{ matrix.dagster }}
         run: uv add --no-sync "ray[all]==${{ matrix.ray }}" "dagster==${{ matrix.dagster }}"
-      - name: Install dependencies
+      - name: Install Default Dependencies
+        run: uv sync
+        env:
+          UV_LOCKED: "1"
+      - name: Check Import With Only Default Dependencies
+        run: uv run python -c "import dagster_ray"
+        env:
+          UV_LOCKED: "1"
+      - name: Install All Dependencies
         run: uv sync --all-extras --dev
       - name: Run tests
         env:
           PYTEST_KUBERAY_VERSIONS: "${{ matrix.kuberay }}"
+          UV_LOCKED: "1"
         run: uv run pytest -v .
 
   lint:
diff --git a/src/dagster_ray/kuberay/resources/base.py b/src/dagster_ray/kuberay/resources/base.py
index 07aa0d6..c6f28de 100644
--- a/src/dagster_ray/kuberay/resources/base.py
+++ b/src/dagster_ray/kuberay/resources/base.py
@@ -1,10 +1,9 @@
 from __future__ import annotations
 
 import re
-from abc import abstractmethod
+from abc import ABC, abstractmethod
 
 from pydantic import Field, PrivateAttr
-from pytest_kubernetes.providers.base import ABC
 
 from dagster_ray._base.constants import DEFAULT_DEPLOYMENT_NAME
 from dagster_ray._base.resources import RayResource