Fix error when using build.rs instead of standard way to trigger code generation

Fix error when using build.rs instead of standard way to trigger code generation #9649