fix class name in module generation