diff --git a/timestream/client_test.go b/timestream/client_test.go
index 3d0f29f..54ddebc 100644
--- a/timestream/client_test.go
+++ b/timestream/client_test.go
@@ -237,7 +237,7 @@ func TestQueryClientRead(t *testing.T) {
 
 	expectedBuildCommand := []*timestreamquery.QueryInput{
 		{
-			QueryString: aws.String(fmt.Sprintf("SELECT * FROM \"%s\".\"%s\" WHERE \"%s\" = '%s' AND quantile != '%s' AND REGEXP_LIKE(job, '%s') AND NOT REGEXP_LIKE(instance, '%s') AND %s BETWEEN FROM_UNIXTIME(%d) AND FROM_UNIXTIME(%d)",
+			QueryString: aws.String(fmt.Sprintf("SELECT * FROM \"%s\".\"%s\" WHERE \"%s\" = '%s' AND \"quantile\" != '%s' AND REGEXP_LIKE(\"job\", '%s') AND NOT REGEXP_LIKE(\"instance\", '%s') AND %s BETWEEN FROM_UNIXTIME(%d) AND FROM_UNIXTIME(%d)",
 				mockDatabaseName, mockTableName, measureNameColumnName, metricName, quantile, jobRegex, instanceRegex, timeColumnName, startUnixInSeconds, endUnixInSeconds)),
 		},
 	}
@@ -270,7 +270,7 @@ func TestQueryClientRead(t *testing.T) {
 	}
 
 	queryInputWithInvalidRegex := &timestreamquery.QueryInput{
-		QueryString: aws.String(fmt.Sprintf("SELECT * FROM \"%s\".\"%s\" WHERE \"%s\" = '%s' AND REGEXP_LIKE(job, '%s') AND %s BETWEEN FROM_UNIXTIME(%d) AND FROM_UNIXTIME(%d)",
+		QueryString: aws.String(fmt.Sprintf("SELECT * FROM \"%s\".\"%s\" WHERE \"%s\" = '%s' AND REGEXP_LIKE(\"job\", '%s') AND %s BETWEEN FROM_UNIXTIME(%d) AND FROM_UNIXTIME(%d)",
 			mockDatabaseName, mockTableName, measureNameColumnName, metricName, invalidRegex, timeColumnName, startUnixInSeconds, endUnixInSeconds)),
 	}