diff --git a/tools/run_tests/xds_k8s_test_driver/tests/gamma/affinity_test.py b/tools/run_tests/xds_k8s_test_driver/tests/gamma/affinity_test.py
index bc89a73cb8737..796049261b472 100644
--- a/tools/run_tests/xds_k8s_test_driver/tests/gamma/affinity_test.py
+++ b/tools/run_tests/xds_k8s_test_driver/tests/gamma/affinity_test.py
@@ -11,8 +11,8 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-import logging
 import datetime
+import logging
 from typing import List, Optional
 
 from absl import flags