diff --git a/src/spectator/dsl/mocks.cr b/src/spectator/dsl/mocks.cr
index 628fe6b..94a41f9 100644
--- a/src/spectator/dsl/mocks.cr
+++ b/src/spectator/dsl/mocks.cr
@@ -38,7 +38,7 @@ module Spectator::DSL
           {{null_double_type_name}}.new(@stubs)
         end
 
-        {% if block %}{{block.body}}{% end %}
+        {{block.body if block}}
       end
 
       {% begin %}
diff --git a/src/spectator/mocks/double.cr b/src/spectator/mocks/double.cr
index ad35dda..b45b9c8 100644
--- a/src/spectator/mocks/double.cr
+++ b/src/spectator/mocks/double.cr
@@ -61,7 +61,7 @@ module Spectator
           end
         {% end %}
 
-        {% if block %}{{block.body}}{% end %}
+        {{block.body if block}}
       end
     end
 
diff --git a/src/spectator/mocks/mock.cr b/src/spectator/mocks/mock.cr
index 901d088..ba72b37 100644
--- a/src/spectator/mocks/mock.cr
+++ b/src/spectator/mocks/mock.cr
@@ -84,7 +84,7 @@ module Spectator
           macro finished
             stub_type {{mocked_type.id}}
 
-            {% if block %}{{block.body}}{% end %}
+            {{block.body if block}}
           end
         end
       {% end %}
@@ -189,7 +189,7 @@ module Spectator
           macro finished
             stub_type {{type_name.id}}
 
-            {% if block %}{{block.body}}{% end %}
+            {{block.body if block}}
           end
         end
       {% end %}