Fix module reference