Update unittest.go