Updating tests that were missed the first time