diff --git a/.gitignore b/.gitignore
index 7ed33a6..416387e 100644
--- a/.gitignore
+++ b/.gitignore
@@ -1,3 +1,4 @@
 .idea
 *.csv
-data
\ No newline at end of file
+data
+__pycache__/
\ No newline at end of file
diff --git a/pyproject.toml b/pyproject.toml
index 8ce0c9d..6eee8a1 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -8,7 +8,7 @@ authors = [
 ]
 requires-python = ">=3.12"
 dependencies = [
-    "flyte==2.0.0b26",
+    "flyte@git+https://github.com/flyteorg/flyte-sdk.git@7fb7d693c99ae44fc720ee78ad11ad533c5447de",
 ]
 
 [build-system]
diff --git a/src/uv_project_example/__pycache__/__init__.cpython-312.pyc b/src/uv_project_example/__pycache__/__init__.cpython-312.pyc
index 2211b98..3b713fb 100644
Binary files a/src/uv_project_example/__pycache__/__init__.cpython-312.pyc and b/src/uv_project_example/__pycache__/__init__.cpython-312.pyc differ
diff --git a/src/uv_project_example/library_a/__pycache__/__init__.cpython-312.pyc b/src/uv_project_example/library_a/__pycache__/__init__.cpython-312.pyc
index 0d72072..23b3971 100644
Binary files a/src/uv_project_example/library_a/__pycache__/__init__.cpython-312.pyc and b/src/uv_project_example/library_a/__pycache__/__init__.cpython-312.pyc differ
diff --git a/src/uv_project_example/library_a/__pycache__/utils.cpython-312.pyc b/src/uv_project_example/library_a/__pycache__/utils.cpython-312.pyc
index 4b6c5d8..d5d7370 100644
Binary files a/src/uv_project_example/library_a/__pycache__/utils.cpython-312.pyc and b/src/uv_project_example/library_a/__pycache__/utils.cpython-312.pyc differ
diff --git a/src/uv_project_example/main_wfs.py b/src/uv_project_example/main_wfs.py
index e1289bb..0666e82 100644
--- a/src/uv_project_example/main_wfs.py
+++ b/src/uv_project_example/main_wfs.py
@@ -1,15 +1,18 @@
 import flyte
 from flyte import Image
 
-from src.uv_project_example.library_a.utils import some_test
+from uv_project_example.library_a.utils import some_test
 
-from src.core.config import ROOT_DIR
+from core.config import ROOT_DIR
 
 image = (
-    Image.from_debian_base(python_version=(3, 12))
+    Image.from_base("python:3.12-slim-bookworm").clone(name="flyte")
+    .with_apt_packages("git")
     .with_apt_packages("ca-certificates", "build-essential")
     .with_uv_project(pyproject_file=ROOT_DIR / "pyproject.toml")
-    # .with_source_folder(ROOT_DIR / "src" / "data_loading")
+    .with_commands("mkdir -p ./src/data_loading")
+    .with_source_folder(ROOT_DIR / "src" / "data_loading", dst="./src/data_loading")
+    .with_pip_packages("git+https://github.com/flyteorg/flyte-sdk.git@7fb7d693c99ae44fc720ee78ad11ad533c5447de")
 )
 
 
@@ -39,6 +42,6 @@ async def t1(data: str = "hello") -> str:
 if __name__ == "__main__":
     # Works with and without root_dir
     flyte.init_from_config(root_dir=ROOT_DIR)  # should we make this work?
-    run = flyte.with_runcontext(mode="remote", copy_style="all").run(t1, data="world")
+    run = flyte.with_runcontext(mode="remote").run(t1, data="world")
     print(run.name)
     print(run.url)
diff --git a/uv.lock b/uv.lock
index 0c17589..9158c3f 100644
--- a/uv.lock
+++ b/uv.lock
@@ -209,8 +209,8 @@ wheels = [
 
 [[package]]
 name = "flyte"
-version = "2.0.0b26"
-source = { registry = "https://pypi.org/simple" }
+version = "2.0.0b30.dev3+g7fb7d693c"
+source = { git = "https://github.com/flyteorg/flyte-sdk.git?rev=7fb7d693c99ae44fc720ee78ad11ad533c5447de#7fb7d693c99ae44fc720ee78ad11ad533c5447de" }
 dependencies = [
     { name = "aiofiles" },
     { name = "aiolimiter" },
@@ -233,9 +233,6 @@ dependencies = [
     { name = "rich-click" },
     { name = "toml" },
 ]
-wheels = [
-    { url = "https://files.pythonhosted.org/packages/ff/9c/4ce5bee1a6bca2f43c0a589eda2ae7c8bb31af63dec831eb669ea70aea8f/flyte-2.0.0b26-py3-none-any.whl", hash = "sha256:d000947d7455b946277b9709f9e7b14933389ea4c1e9c484df82acbbd16ea05d", size = 381510, upload-time = "2025-10-29T22:27:12.218Z" },
-]
 
 [[package]]
 name = "flyteidl"
@@ -253,17 +250,14 @@ wheels = [
 
 [[package]]
 name = "flyteidl2"
-version = "2.0.0a11"
-source = { registry = "https://pypi.org/simple" }
+version = "2.0.0a14.dev3+g91ba87d5e"
+source = { git = "https://github.com/flyteorg/flyte.git?subdirectory=gen%2Fpython&rev=v2#91ba87d5e4d0a0b99706d87a568c7cbf2b94cb21" }
 dependencies = [
     { name = "googleapis-common-protos" },
     { name = "protobuf" },
     { name = "protoc-gen-openapiv2" },
     { name = "protovalidate" },
 ]
-wheels = [
-    { url = "https://files.pythonhosted.org/packages/3e/83/ed4e62796135abd41d4440c220e8390b7dc026f7342c92555ac06a52da83/flyteidl2-2.0.0a11-py3-none-any.whl", hash = "sha256:17d72b25ab3aec21aef9d5b0cacf936560e25f4b898c196650e9c5a9259f5ff1", size = 192937, upload-time = "2025-10-29T22:04:42.92Z" },
-]
 
 [[package]]
 name = "fsspec"
@@ -982,4 +976,4 @@ dependencies = [
 ]
 
 [package.metadata]
-requires-dist = [{ name = "flyte", specifier = "==2.0.0b26" }]
+requires-dist = [{ name = "flyte", git = "https://github.com/flyteorg/flyte-sdk.git?rev=7fb7d693c99ae44fc720ee78ad11ad533c5447de" }]