diff --git a/research/GDP_2019/imdb_tutorial.py b/research/GDP_2019/imdb_tutorial.py
index 03a1a34..cc9ee42 100644
--- a/research/GDP_2019/imdb_tutorial.py
+++ b/research/GDP_2019/imdb_tutorial.py
@@ -27,9 +27,8 @@ import tensorflow as tf
 from keras.preprocessing import sequence
 
 from tensorflow_privacy.privacy.optimizers import dp_optimizer
-
-from tensorflow_privacy.privacy.analysis.gdp_accountant import *
-
+from tensorflow_privacy.privacy.analysis.gdp_accountant import compute_eps_poisson
+from tensorflow_privacy.privacy.analysis.gdp_accountant import compute_mu_poisson
 
 #### FLAGS
 FLAGS = flags.FLAGS