Fix module problem