fixed problem in test