Dumb fix for module reference