diff --git a/.gitignore b/.gitignore
index fb61326..28b87b6 100644
--- a/.gitignore
+++ b/.gitignore
@@ -12,4 +12,7 @@
 
 # Haskell IR files
 *.hi
-*.o
\ No newline at end of file
+*.o
+
+# Graphing images created with Cairo
+*.png
diff --git a/flake.nix b/flake.nix
index c3c22ff..dd80948 100644
--- a/flake.nix
+++ b/flake.nix
@@ -65,7 +65,7 @@
             find . -type f -name '*.o' -delete
 
           # Delete any test graphs created
-            find . -type f -name 'test.png' -delete
+            find . -type f -name '*.png' -delete
         '';
       in
       pkgs.mkShell {