diff --git a/DynamoDbEncryption/Makefile b/DynamoDbEncryption/Makefile
index 1ad542b3e..d79f599bf 100644
--- a/DynamoDbEncryption/Makefile
+++ b/DynamoDbEncryption/Makefile
@@ -165,12 +165,12 @@ STRUCTUREDENCRYPTION_INDEX_FILE_WITH_EXTERN_STRING="module {:extern \"software.a
 STRUCTUREDENCRYPTION_INDEX_FILE_WITHOUT_EXTERN_STRING="module StructuredEncryption"
 
 # Target to restore all directories in a list
-# TODO: Remove this once we don't copy all of the directories into implementation and test https://sim.amazon.com/issues/CrypTool-5447
-RESTORE_DIRS := README.md
-_polymorph_go: restore_directories
-restore_directories:
-	@for dir in $(RESTORE_DIRS); do \
-		if [ -d "runtimes/go/ImplementationFromDafny-go/$$dir" ]; then \
+# TODO: Remove this once we don't copy all of the directories and files into implementation and test https://sim.amazon.com/issues/CrypTool-5447
+RESTORE_FILES_DIR := README.md
+_polymorph_go: restore_file_directories
+restore_file_directories:
+	@for dir in $(RESTORE_FILES_DIR); do \
+		if [ -e "runtimes/go/ImplementationFromDafny-go/$$dir" ]; then \
 			cp -Rf runtimes/go/ImplementationFromDafny-go/$$dir runtimes/go/; \
 			rm -rf runtimes/go/ImplementationFromDafny-go/$$dir; \
 			rm -rf runtimes/go/TestsFromDafny-go/$$dir; \