Fix: use the right parameter type