Updating my test code...