diff --git a/create_requirement_images.py b/create_requirement_images.py
index 99ae8b7..d5cf491 100755
--- a/create_requirement_images.py
+++ b/create_requirement_images.py
@@ -227,8 +227,8 @@ def make_header(position, project_files, files_and_libs):
         for cur_file in project_files:
             # string for individual file
             if isinstance(cur_file, str):
-                if "." in cur_file[-5:]:
-                    cur_extension = cur_file.split(".")[-1]
+                if "." in cur_file:
+                    cur_extension = cur_file.rsplit(".", 1)[-1]
                     if cur_extension in SHOWN_FILETYPES:
                         project_files_to_draw.append(cur_file)
             # tuple for directory