diff --git a/waves.py b/waves.py
index e426d6d..21d6fce 100755
--- a/waves.py
+++ b/waves.py
@@ -104,8 +104,8 @@ def create_wpotd(output):
         ctx.line_to(*points[0])
         ctx.fill()
         lastpoints = points
-    return surface
-    #surface.write_to_png(output)  # Output to PNG
+    #return surface
+    surface.write_to_png(output)  # Output to PNG
 
 def main():
     create_wpotd('out/waves.png')