Fix module import error