fix code gen test case