Updates a unit test to remove references to code that no longer exists.