Fix: use new method