Fix regression with categories