Fix module imports