Change the add_action for theme_setup function to the correct hook