Fix build as a result of kryo upgrade