diff --git a/run_integration_tests.py b/run_integration_tests.py
index 64e5b856..e8abf4f4 100755
--- a/run_integration_tests.py
+++ b/run_integration_tests.py
@@ -5,7 +5,7 @@ import logging
 import unittest
 
 from argparse import ArgumentParser
-from distutils.core import Command
+from setuptools import Command
 from importlib import import_module
 from pathlib import Path
 
diff --git a/run_tests.py b/run_tests.py
index f5103b15..62055e40 100755
--- a/run_tests.py
+++ b/run_tests.py
@@ -5,7 +5,7 @@ import logging
 import unittest
 
 from argparse import ArgumentParser
-from distutils.core import Command
+from setuptools import Command
 from importlib import import_module
 from pathlib import Path