diff --git a/tests/contrib/networks/test_tf_ctx_feed.py b/tests/contrib/networks/test_tf_ctx_feed.py
index 9690dec5..e8ee789b 100755
--- a/tests/contrib/networks/test_tf_ctx_feed.py
+++ b/tests/contrib/networks/test_tf_ctx_feed.py
@@ -58,7 +58,7 @@ def run_feeding(network, network_config, create_minibatch_func, logger,
         init_config(network_config)
         # Init network.
         network.compile(config=network_config, reset_graph=True, graph_seed=42)
-        minibatch = create_minibatch_func(config=network_config, labels_count=labels_scaler)
+        minibatch = create_minibatch_func(config=network_config, labels_scaler=labels_scaler)
 
         network_optimiser = network_config.Optimiser.minimize(network.Cost)