Fix module bugs