Update suite_test.go