Fix for missing module issue