diff --git a/samples/py_native/pb_project.yaml b/samples/py_native/pb_project.yaml
index d9475721..822dd927 100644
--- a/samples/py_native/pb_project.yaml
+++ b/samples/py_native/pb_project.yaml
@@ -5,6 +5,6 @@ model_folders:
   - models
 packages:
   - name: feature_table
-    url: https://github.com/rudderlabs/profiles-features-integration-test/tree/feat/add_diff_timestamps
+    url: https://github.com/rudderlabs/profiles-features-integration-test
 python_requirements:
   - profiles_mlcorelib