diff --git a/src/snowflake/cli/_plugins/streamlit/streamlit_entity.py b/src/snowflake/cli/_plugins/streamlit/streamlit_entity.py
index 6b187ba54b..8a364c5517 100644
--- a/src/snowflake/cli/_plugins/streamlit/streamlit_entity.py
+++ b/src/snowflake/cli/_plugins/streamlit/streamlit_entity.py
@@ -10,7 +10,7 @@
     StreamlitEntityModel,
 )
 from snowflake.cli.api.entities.common import EntityBase
-from snowflake.cli.api.project.schemas.v1.native_app.path_mapping import PathMapping
+from snowflake.cli.api.project.schemas.entities.common import PathMapping
 
 
 # WARNING: This entity is not implemented yet. The logic below is only for demonstrating the