diff --git a/python_modules/libraries/dagster-powerbi/dagster_powerbi_tests/test_translator.py b/python_modules/libraries/dagster-powerbi/dagster_powerbi_tests/test_translator.py
index 155b113c6a500..a9d06118f75be 100644
--- a/python_modules/libraries/dagster-powerbi/dagster_powerbi_tests/test_translator.py
+++ b/python_modules/libraries/dagster-powerbi/dagster_powerbi_tests/test_translator.py
@@ -1,7 +1,7 @@
 from dagster._core.definitions.asset_key import AssetKey
 from dagster._core.definitions.asset_spec import AssetSpec
 from dagster_powerbi import DagsterPowerBITranslator
-from dagster_powerbi.translator import PowerBIWorkspaceData
+from dagster_powerbi.translator import PowerBIContentData, PowerBIWorkspaceData
 
 
 def test_translator_dashboard_spec(workspace_data: PowerBIWorkspaceData) -> None: