Fix test_codegen