Fixed module issue