fixed code generator test