Add return statement to redirect_with_message