fixing bug in test