1 files changed, 1 insertions, 2 deletions
diff --git a/src/lex.l b/src/lex.l
index 02bd2c9..61b5d7a 100644
--- a/src/lex.l
+++ b/src/lex.l
@@ -8,7 +8,6 @@
 #include "string.h"
 int saved_state = 0;
 extern char *input_filename;
-extern long int gen_level;
 
 %}
 %x BLOCK_COMMENT
@@ -146,7 +145,7 @@ yylval.integer = BRW_CHANNEL_W;
  /* send argument tokens */
 "mlen" { return MSGLEN; }
 "rlen" { return RETURNLEN; }
-"math" { if (gen_level >= 6) { yylval.integer = BRW_OPCODE_MATH; return MATH_INST; } else return MATH; }
+"math" { if (IS_GENp(6)) { yylval.integer = BRW_OPCODE_MATH; return MATH_INST; } else return MATH; }
 "sampler" { return SAMPLER; }
 "gateway" { return GATEWAY; }
 "read" { return READ; }