Fixed problems with my test methods