small modification to correct test bug