diff --git a/UnitySDK/Assets/ML-Agents/Examples/Tennis/Scripts/TennisAgent.cs b/UnitySDK/Assets/ML-Agents/Examples/Tennis/Scripts/TennisAgent.cs
index d99ce957d2..b511630062 100755
--- a/UnitySDK/Assets/ML-Agents/Examples/Tennis/Scripts/TennisAgent.cs
+++ b/UnitySDK/Assets/ML-Agents/Examples/Tennis/Scripts/TennisAgent.cs
@@ -26,7 +26,7 @@ public class TennisAgent : Agent
     public override void InitializeAgent()
     {
         agentRb = GetComponent<Rigidbody>();
-        ballRb = GetComponent<Rigidbody>();
+        ballRb = ball.GetComponent<Rigidbody>();
         var canvas = GameObject.Find(CanvasName);
         GameObject scoreBoard;
         if (invertX)
diff --git a/UnitySDK/Assets/ML-Agents/Examples/Tennis/TFModels/TennisLearning.nn b/UnitySDK/Assets/ML-Agents/Examples/Tennis/TFModels/TennisLearning.nn
index c443d2ca84..31af92119b 100644
Binary files a/UnitySDK/Assets/ML-Agents/Examples/Tennis/TFModels/TennisLearning.nn and b/UnitySDK/Assets/ML-Agents/Examples/Tennis/TFModels/TennisLearning.nn differ
diff --git a/UnitySDK/Assets/ML-Agents/Examples/Tennis/TFModels/TennisLearning.nn.meta b/UnitySDK/Assets/ML-Agents/Examples/Tennis/TFModels/TennisLearning.nn.meta
index 861f9ffbba..5487b616dd 100644
--- a/UnitySDK/Assets/ML-Agents/Examples/Tennis/TFModels/TennisLearning.nn.meta
+++ b/UnitySDK/Assets/ML-Agents/Examples/Tennis/TFModels/TennisLearning.nn.meta
@@ -1,5 +1,5 @@
 fileFormatVersion: 2
-guid: 14a6405c810e241918d14d3b6226b87b
+guid: 0b22bc7e45e8347fd897bdf0e1d38c6e
 ScriptedImporter:
   userData: 
   assetBundleName: