diff --git a/changes/256.fixed b/changes/256.fixed
new file mode 100644
index 00000000..fd3b5588
--- /dev/null
+++ b/changes/256.fixed
@@ -0,0 +1 @@
+Patch to make the generated Dockerfile compatible with kaniko behaviour on AWS.
diff --git a/substrafl/remote/register/register.py b/substrafl/remote/register/register.py
index 55745fb8..51bb7544 100644
--- a/substrafl/remote/register/register.py
+++ b/substrafl/remote/register/register.py
@@ -84,7 +84,7 @@
 RUN python{python_version} -m pip install --no-cache-dir -r requirements.txt
 
 USER root
-RUN apt-get purge -y --auto-remove build-essential *-dev
+RUN apt-get update -y && apt-get purge -y --auto-remove build-essential *-dev
 USER user
 
 # Copy all other files
diff --git a/tests/remote/register/test_register.py b/tests/remote/register/test_register.py
index 8ad17e70..6c22fede 100644
--- a/tests/remote/register/test_register.py
+++ b/tests/remote/register/test_register.py
@@ -124,7 +124,7 @@ def test_create_dockerfile(tmp_path, local_installable_module):
 RUN python{python_version} -m pip install --no-cache-dir -r requirements.txt
 
 USER root
-RUN apt-get purge -y --auto-remove build-essential *-dev
+RUN apt-get update -y && apt-get purge -y --auto-remove build-essential *-dev
 USER user
 
 # Copy all other files