diff --git a/mini_proj/test_nn.py b/mini_proj/test_nn.py
index f33f837..0d8ec2e 100644
--- a/mini_proj/test_nn.py
+++ b/mini_proj/test_nn.py
@@ -68,5 +68,4 @@ if __name__ == '__main__':
 
     # Stitch image TODO!
     # Show image
-    cv.imshow('Image', image)
-    cv.waitKey(0)
+    cv2.imwrite('output.png',image)