Removed hardcoding of code gen.