Updated Razor engine to use right code generator