Fixed module use issue