Fix reference to incorrect argument