diff --git a/OpenRA.Game/OpenRA.Game.csproj b/OpenRA.Game/OpenRA.Game.csproj
index 82eaf8922b..a3c6b6a510 100755
--- a/OpenRA.Game/OpenRA.Game.csproj
+++ b/OpenRA.Game/OpenRA.Game.csproj
@@ -82,7 +82,6 @@
     <Compile Include="Group.cs" />
     <Compile Include="Orders\GenericSelectTarget.cs" />
     <Compile Include="Traits\Activities\Leap.cs" />
-    <Compile Include="Traits\AI\ReturnOnIdle.cs" />
     <Compile Include="Traits\Attack\AttackLeap.cs" />
     <Compile Include="Traits\DetectCloaked.cs" />
     <Compile Include="Traits\Player\ActorGroupProxy.cs" />
diff --git a/OpenRA.Game/Traits/Activities/ReturnToBase.cs b/OpenRA.Game/Traits/Activities/ReturnToBase.cs
index 349e2357ed..38ff915767 100644
--- a/OpenRA.Game/Traits/Activities/ReturnToBase.cs
+++ b/OpenRA.Game/Traits/Activities/ReturnToBase.cs
@@ -24,7 +24,7 @@ using OpenRA.GameRules;
 
 namespace OpenRA.Traits.Activities
 {
-	class ReturnToBase : IActivity
+	public class ReturnToBase : IActivity
 	{
 		public IActivity NextActivity { get; set; }
 
diff --git a/OpenRA.Mods.RA/OpenRA.Mods.RA.csproj b/OpenRA.Mods.RA/OpenRA.Mods.RA.csproj
index 8a4843be42..fe096846ee 100644
--- a/OpenRA.Mods.RA/OpenRA.Mods.RA.csproj
+++ b/OpenRA.Mods.RA/OpenRA.Mods.RA.csproj
@@ -103,6 +103,7 @@
     <Compile Include="RenderSpy.cs" />
     <Compile Include="RepairableNear.cs" />
     <Compile Include="Crates\SpeedUpgradeCrateAction.cs" />
+    <Compile Include="ReturnOnIdle.cs" />
     <Compile Include="SelfHealing.cs" />
     <Compile Include="SonarPulsePower.cs" />
     <Compile Include="Spy.cs" />
diff --git a/OpenRA.Game/Traits/AI/ReturnOnIdle.cs b/OpenRA.Mods.RA/ReturnOnIdle.cs
similarity index 93%
rename from OpenRA.Game/Traits/AI/ReturnOnIdle.cs
rename to OpenRA.Mods.RA/ReturnOnIdle.cs
index efd8f16075..1484bb7899 100644
--- a/OpenRA.Game/Traits/AI/ReturnOnIdle.cs
+++ b/OpenRA.Mods.RA/ReturnOnIdle.cs
@@ -21,8 +21,9 @@
 using System;
 using OpenRA.Traits.Activities;
 using System.Drawing;
+using OpenRA.Traits;
 
-namespace OpenRA.Traits.AI
+namespace OpenRA.Mods.RA
 {
 	class ReturnOnIdleInfo : TraitInfo<ReturnOnIdle> { }