fixed the tests.. missing some obvious stuff