Fix problem with source code generation