test simplified
made the tests simpler