Added fix to make code generator work as per a bug in Modisco 0.11.0.