Out of date unit test fix. I might stop committing things now.