Fixing tests. Still getting one with the update user flash success though.