Fix module initialization