Update tests, but they aren't working