fixed problem with test