diff --git a/rllib/contrib/bandits/examples/tune_LinUCB_train_recommendation.py b/rllib/contrib/bandits/examples/tune_LinUCB_train_recommendation.py
index c27f047bc..6b3c6daa7 100644
--- a/rllib/contrib/bandits/examples/tune_LinUCB_train_recommendation.py
+++ b/rllib/contrib/bandits/examples/tune_LinUCB_train_recommendation.py
@@ -27,7 +27,7 @@ if __name__ == "__main__":
 
     start_time = time.time()
     analysis = tune.run(
-        LinUCBTrainer,
+        "contrib/LinUCB",
         config=UCB_CONFIG,
         stop={"training_iteration": training_iterations},
         num_samples=5,