fixed issue with test