Fix: use correct variable