Fix module import bug