diff --git a/heat-config-docker-compose/install.d/hook-docker-compose.py b/heat-config-docker-compose/install.d/hook-docker-compose.py
index f73332e..a73ec27 100755
--- a/heat-config-docker-compose/install.d/hook-docker-compose.py
+++ b/heat-config-docker-compose/install.d/hook-docker-compose.py
@@ -98,7 +98,7 @@ def main(argv=sys.argv):
             for i in ast.literal_eval(input_values.get('env_files')))
 
     for file in compose_env_files:
-        if file in input_env_files.keys():
+        if file in input_env_files:
             write_input_file(file, input_env_files.get(file))
 
     cmd = [
diff --git a/heat-config-json-file/install.d/hook-json-file.py b/heat-config-json-file/install.d/hook-json-file.py
index 84c13f8..a029135 100755
--- a/heat-config-json-file/install.d/hook-json-file.py
+++ b/heat-config-json-file/install.d/hook-json-file.py
@@ -34,7 +34,7 @@ def main(argv=sys.argv):
 
     c = json.load(sys.stdin)['config']
 
-    for fname in c.keys():
+    for fname in c:
         prepare_dir(os.path.dirname(fname))
         data = c.get(fname)
         with open(fname, 'w') as json_data_file: