ticket:7 Fix test