Update data_test.go