Add missing module reload