diff --git a/lib/mix_tasks.ex b/lib/mix_tasks.ex
index efe53d7..add3140 100644
--- a/lib/mix_tasks.ex
+++ b/lib/mix_tasks.ex
@@ -131,16 +131,12 @@ defmodule Mix.Tasks.Aoc do
       p2 = mod.part2(parsed)
 
       result = fn (part, got, expected) ->
-        if expected do
-          if stringify(got) != expected do
-            IO.puts(IO.ANSI.format [:red, "Test #{test}.#{part} failed: expected #{expected}"])
-            dbg(got)
-          else
-            IO.puts(IO.ANSI.format [:green, "Test #{test}.#{part} succeeded"])
-          end
-        else
-          IO.puts("Test #{test}.#{part}:")
-          dbg(got)
+        cond do
+          stringify(got) == expected ->
+            IO.puts(IO.ANSI.format [:green, "Test #{test}.#{part} succeeded (#{expected})"])
+          expected ->
+            IO.puts(IO.ANSI.format [:red, "Test #{test}.#{part} failed: expected #{expected}, got #{stringify(got)}"])
+          true ->  IO.puts("Test #{test}.#{part}: #{stringify(got)}")
         end
       end