Add missing module reference