Updating tests that were in another file