fix after_initialzie support