fix bug with test