diff --git a/src/ldtk/macro/TypeBuilder.hx b/src/ldtk/macro/TypeBuilder.hx
index fdf0d33..be53b34 100644
--- a/src/ldtk/macro/TypeBuilder.hx
+++ b/src/ldtk/macro/TypeBuilder.hx
@@ -1,6 +1,6 @@
 package ldtk.macro;
 
-#if( !macro && !display )
+#if( !macro && !haxe.macro.Context.defined("display") )
 #error "This class should not be used outside of macros"
 #end