keep on siplifying tests
simplified code