diff --git a/bob/learn/tensorflow/test/test_db_to_tfrecords.py b/bob/learn/tensorflow/test/test_db_to_tfrecords.py
index 2318ca591c01c8d276940b5cbfc1359bab9c9c0d..db5491d08066aa5e5138b447d7e32677576e1900 100644
--- a/bob/learn/tensorflow/test/test_db_to_tfrecords.py
+++ b/bob/learn/tensorflow/test/test_db_to_tfrecords.py
@@ -107,7 +107,7 @@ def test_datasets_to_tfrecords():
     runner = CliRunner()
     with runner.isolated_filesystem():
         output_path = './test'
-        args = (dummy_config, '--outputs', output_path)
+        args = (dummy_config, '--output', output_path)
         result = runner.invoke(
             datasets_to_tfrecords, args=args, standalone_mode=False)
         assert_click_runner_result(result)
diff --git a/bob/learn/tensorflow/test/test_utils.py b/bob/learn/tensorflow/test/test_utils.py
index e4f61241acb9a34503f88f6567075c0b4e4fcdfa..e08b5311581c25f46dccc47b78a8a73549ab5142 100644
--- a/bob/learn/tensorflow/test/test_utils.py
+++ b/bob/learn/tensorflow/test/test_utils.py
@@ -3,8 +3,8 @@
 # @author: Tiago de Freitas Pereira <tiago.pereira@idiap.ch>
 
 import numpy
-from bob.learn.tensorflow.utils import compute_embedding_accuracy, cdist,\
-     compute_embedding_accuracy_tensors, predict_using_tensors
+from bob.learn.tensorflow.utils import compute_embedding_accuracy, \
+     compute_embedding_accuracy_tensors
 
 import tensorflow as tf
 """