ticket:10. fix tests