FIX to Sorbet migration