Fix: change function declaration