Fix module setup