Fix function call for test