Fixed missing module issue