More fixes

Author: penelopeysm

main.jl

@@ -49,14 +49,14 @@ elseif length(ARGS) == 3 && ARGS[1] == "--run"
     catch e
         if e isa ADIncorrectException
             # First check for completely incorrect ones
-            for (a, b) in zip(result.grad_expected, result.grad_actual)
+            for (a, b) in zip(e.grad_expected, e.grad_actual)
                 if !isnan(a) && !isnan(b) && abs(a - b) > 1e-6
                     println("wrong")
                     exit()
                 end
             end
             # If not, check for NaN's and report those
-            if any(isnan, result.grad_expected) || any(isnan, result.grad_actual)
+            if any(isnan, e.grad_expected) || any(isnan, e.grad_actual)
                 println("NaN")
             else
                 # Something else went wrong, shouldn't happen