Removed the need for the after_initialize hook