fix bug tests