re-fix after_initialize support