Module import bug fix