diff --git a/src/main/java/uk/ac/rhul/cs/javatester/TesterOutput.java b/src/main/java/uk/ac/rhul/cs/javatester/TesterOutput.java
index c80fed12e61821e051c5108497a5494b196937c5..48d799a83b28a15803513680b39630e23b22b5bd 100644
--- a/src/main/java/uk/ac/rhul/cs/javatester/TesterOutput.java
+++ b/src/main/java/uk/ac/rhul/cs/javatester/TesterOutput.java
@@ -48,7 +48,8 @@ public class TesterOutput {
                 indented = CODE_INDENT + code;
             } else if (code.length() > 0) {
                 indented
-                    = CODE_INDENT + code.replaceAll("\n(?!$)", "\n" + CODE_INDENT);
+                    = CODE_INDENT +
+                      code.replaceAll("\n(?!($|\n))", "\n" + CODE_INDENT);
             }
         }