Updating CSRF tests to match new API