Fix regression