diff --git a/Directory.Build.props b/Directory.Build.props
index a96cf011..11390a0b 100644
--- a/Directory.Build.props
+++ b/Directory.Build.props
@@ -14,7 +14,7 @@
     <PackageIcon>icon.png</PackageIcon>
     <IncludeSymbols>true</IncludeSymbols>
     <RepositoryType>git</RepositoryType>
-    <VersionPrefix>0.3.0</VersionPrefix>
+    <VersionPrefix>0.3.1</VersionPrefix>
     <VersionSuffix></VersionSuffix>
     <LangVersion>12.0</LangVersion>
   </PropertyGroup>
diff --git a/src/Bonsai.ML.LinearDynamicalSystems.Design/ForecastImageOverlay.cs b/src/Bonsai.ML.LinearDynamicalSystems.Design/ForecastImageOverlay.cs
index f9593fbc..7ebfb04d 100644
--- a/src/Bonsai.ML.LinearDynamicalSystems.Design/ForecastImageOverlay.cs
+++ b/src/Bonsai.ML.LinearDynamicalSystems.Design/ForecastImageOverlay.cs
@@ -8,7 +8,7 @@
 using OxyPlot;
 
 [assembly: TypeVisualizer(typeof(Bonsai.ML.LinearDynamicalSystems.Design.ForecastImageOverlay),
-    Target = typeof(Bonsai.ML.LinearDynamicalSystems.Kinematics.Forecast))]
+    Target = typeof(MashupSource<ImageMashupVisualizer, Bonsai.ML.LinearDynamicalSystems.Design.ForecastVisualizer>))]
 
 namespace Bonsai.ML.LinearDynamicalSystems.Design
 {
diff --git a/src/Bonsai.ML.LinearDynamicalSystems.Design/ForecastPlotOverlay.cs b/src/Bonsai.ML.LinearDynamicalSystems.Design/ForecastPlotOverlay.cs
index b586476a..a4951708 100644
--- a/src/Bonsai.ML.LinearDynamicalSystems.Design/ForecastPlotOverlay.cs
+++ b/src/Bonsai.ML.LinearDynamicalSystems.Design/ForecastPlotOverlay.cs
@@ -7,7 +7,7 @@
 using OxyPlot.Series;
 using OxyPlot;
 
-[assembly: TypeVisualizer(typeof(Bonsai.ML.LinearDynamicalSystems.Design.ForecastImageOverlay),
+[assembly: TypeVisualizer(typeof(Bonsai.ML.LinearDynamicalSystems.Design.ForecastPlotOverlay),
     Target = typeof(MashupSource<Bonsai.ML.LinearDynamicalSystems.Design.KinematicStateVisualizer, Bonsai.ML.LinearDynamicalSystems.Design.ForecastVisualizer>))]
 
 namespace Bonsai.ML.LinearDynamicalSystems.Design