Change the action hook to `after_setup_theme`