fixed problem with test.