diff --git a/projects/rocprofiler-compute/tests/test_analyze_commands.py b/projects/rocprofiler-compute/tests/test_analyze_commands.py
index e79cdfef9b..fb6ebe075d 100644
--- a/projects/rocprofiler-compute/tests/test_analyze_commands.py
+++ b/projects/rocprofiler-compute/tests/test_analyze_commands.py
@@ -2,9 +2,9 @@ import os.path
 from pathlib import Path
 from unittest.mock import patch
 import pytest
-import imp
+from importlib.machinery import SourceFileLoader
 
-omniperf = imp.load_source("omniperf", "src/omniperf")
+omniperf = SourceFileLoader("omniperf", "src/omniperf").load_module()
 
 
 def test_valid_path_mi100():
diff --git a/projects/rocprofiler-compute/tests/test_analyze_workloads.py b/projects/rocprofiler-compute/tests/test_analyze_workloads.py
index cfe23f076a..20e5ecc1bd 100644
--- a/projects/rocprofiler-compute/tests/test_analyze_workloads.py
+++ b/projects/rocprofiler-compute/tests/test_analyze_workloads.py
@@ -1,8 +1,9 @@
 import pytest
 from unittest.mock import patch
-import imp
+from importlib.machinery import SourceFileLoader
+
+omniperf = SourceFileLoader("omniperf", "src/omniperf").load_module()
 
-omniperf = imp.load_source("omniperf", "src/omniperf")
 ##################################################
 ##          Generated tests                     ##
 ##################################################
diff --git a/projects/rocprofiler-compute/tests/test_import_workloads.py b/projects/rocprofiler-compute/tests/test_import_workloads.py
index 5a0f2f11f3..7b169f8a9d 100644
--- a/projects/rocprofiler-compute/tests/test_import_workloads.py
+++ b/projects/rocprofiler-compute/tests/test_import_workloads.py
@@ -1,8 +1,8 @@
 import pytest
 from unittest.mock import patch
-import imp
+from importlib.machinery import SourceFileLoader
 
-omniperf = imp.load_source("omniperf", "src/omniperf")
+omniperf = SourceFileLoader("omniperf", "src/omniperf").load_module()
 
 ##################################################
 ##          Generated tests                     ##