Fixed issue with missing module