Trying to fix test, but failing.