diff --git a/extended/code_update/code_update_utils.robot b/extended/code_update/code_update_utils.robot
index 3a4df5f..c466019 100755
--- a/extended/code_update/code_update_utils.robot
+++ b/extended/code_update/code_update_utils.robot
@@ -104,7 +104,7 @@
     ...  ls -lh ${targ_file_path}
     ${status}=  Run Keyword And Return Status
     ...  Should Contain  ${file_size}  32.0M  msg=Incomplete file transfer.
-    [return]  ${status}
+    RETURN  ${status}
 
 
 Check If File Exist
diff --git a/extended/code_update/host_code_update.robot b/extended/code_update/host_code_update.robot
index 502a821..728a80f 100644
--- a/extended/code_update/host_code_update.robot
+++ b/extended/code_update/host_code_update.robot
@@ -287,7 +287,7 @@
 
     ${version}= BMC Execute Command
     ...  grep extended_version= ${manifest_path}
-    [return] ${version.split(",")}
+    RETURN   ${version.split(",")}
 
 
 Code Update Test Teardown