Fixed my test mistakes