Update functions_test.go