Minor fix. Misses naming update.