Fix module import