Fix module class