Fix tests for function_refs