Fix module name for class block