Update tests to include function calls