diff --git a/tensorflow_privacy/privacy/dp_query/gaussian_query.py b/tensorflow_privacy/privacy/dp_query/gaussian_query.py
index b50fae6f3..834abccc0 100644
--- a/tensorflow_privacy/privacy/dp_query/gaussian_query.py
+++ b/tensorflow_privacy/privacy/dp_query/gaussian_query.py
@@ -15,7 +15,7 @@
 
 import collections
 import dp_accounting
-import packaging
+import packaging.version
 import tensorflow as tf
 from tensorflow_privacy.privacy.dp_query import dp_query