diff --git a/src/main.lua b/src/main.lua
index 27d8f6e..f4c0fda 100644
--- a/src/main.lua
+++ b/src/main.lua
@@ -96,7 +96,7 @@ local function newdesign()
                     term.blit(" ", colors.toBlit(colors.white), colors.toBlit(colors.white))
                 elseif currentbuf == 2 then -- no ceiling
                     term.blit(" ", colors.toBlit(colors.red), colors.toBlit(colors.red))
-                elseif currentbuf == 0 then -- nothing
+                elseif currentbuf == 0 or currentbuf == nil then -- nothing
                     term.blit("\127", colors.toBlit(colors.gray), colors.toBlit(colors.black))
                 end
             end