Fix a bug from putting the code in a module.