Fixed module type naming

This PR fixes the type of the module to ensure that it gets imported successfully.