Fix regression in Response generation