Fix module issue