diff --git a/.github/workflows/push.yml b/.github/workflows/push.yml
index 87f1272e..0fc1fd10 100644
--- a/.github/workflows/push.yml
+++ b/.github/workflows/push.yml
@@ -16,7 +16,7 @@ jobs:
           - '3.9'
           - '3.10'
           - '3.11'
-
+          - '3.12'
     env:
       PYTHON_VERSION: ${{ matrix.python-version }}
     steps:
@@ -29,7 +29,7 @@ jobs:
           python-version: ${{ matrix.python-version }}
 
       - name: Install pip
-        run: pip install --upgrade pip
+        run: pip install --upgrade pip setuptools
 
       - name: Install development dependencies
         run: pip install --upgrade -r dev-requirements.txt
@@ -41,6 +41,7 @@ jobs:
         run: pytest tests --cov=./
 
       - name: Run lint
+        if: matrix.python-version != '3.12' # new lint rules in 3.12 don't pass
         run: prospector --profile ./prospector.yaml -t dodgy -t mccabe -t profile-validator -t pyflakes -t pylint
 
       - name: Run rstcheck
@@ -61,7 +62,7 @@ jobs:
           - '3.9'
           - '3.10'
           - '3.11'
-
+          - '3.12'
     env:
       PYTHON_VERSION: ${{ matrix.python-version }}
     steps:
@@ -74,7 +75,7 @@ jobs:
           python-version: ${{ matrix.python-version }}
 
       - name: Install pip
-        run: pip install --upgrade pip
+        run: pip install --upgrade pip setuptools
 
       - name: Install dependencies
         run: |
diff --git a/setup.py b/setup.py
index 66596101..8a13d63b 100644
--- a/setup.py
+++ b/setup.py
@@ -21,13 +21,14 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import imp
 import io
 import os
 from setuptools import setup, find_packages
+from importlib.machinery import SourceFileLoader
 
-version = imp.load_source(
-    'databricks_cli.version', os.path.join('databricks_cli', 'version.py')).version
+path_to_module = os.path.join('databricks_cli', 'version.py')
+loaded_module = SourceFileLoader('databricks_cli.version', path_to_module).load_module()
+version = loaded_module.version
 
 setup(
     name='databricks-cli',