[PATCH] Fix module import bug