diff --git a/nimpretty/tests/expected/simple2.nim b/nimpretty/tests/expected/simple2.nim
index e265aaff03..99b48e543d 100644
--- a/nimpretty/tests/expected/simple2.nim
+++ b/nimpretty/tests/expected/simple2.nim
@@ -2,3 +2,8 @@
 var x: int = 2
 
 echo x
+
+proc fun*() =
+  echo "ok"
+  ## doc comment
+  # regular comment
diff --git a/nimpretty/tests/simple2.nim b/nimpretty/tests/simple2.nim
index e265aaff03..d07885fc55 100644
--- a/nimpretty/tests/simple2.nim
+++ b/nimpretty/tests/simple2.nim
@@ -2,3 +2,8 @@
 var x: int = 2
 
 echo x
+
+proc fun*()=
+  echo "ok"
+  ## doc comment
+  # regular comment