Fix for new sorbet structure