Fix expected line in test