Fix reference error when using module.