Fix the initialize method