Fix initialize method in extension class