Updated the tests... Really need some help here.