Index: Ringtoets/Common/test/Ringtoets.Common.Forms.Test/Views/GeneralResultFaultTreeIllustrationPointViewTest.cs
===================================================================
diff -u -r88c078b4ce562509a08e6fb962adb8e773f080a7 -rc072c430cf3193c3ada31a81bd0cd4e75f4a068a
--- Ringtoets/Common/test/Ringtoets.Common.Forms.Test/Views/GeneralResultFaultTreeIllustrationPointViewTest.cs	(.../GeneralResultFaultTreeIllustrationPointViewTest.cs)	(revision 88c078b4ce562509a08e6fb962adb8e773f080a7)
+++ Ringtoets/Common/test/Ringtoets.Common.Forms.Test/Views/GeneralResultFaultTreeIllustrationPointViewTest.cs	(.../GeneralResultFaultTreeIllustrationPointViewTest.cs)	(revision c072c430cf3193c3ada31a81bd0cd4e75f4a068a)
@@ -31,6 +31,7 @@
 using Ringtoets.Common.Data.Calculation;
 using Ringtoets.Common.Data.IllustrationPoints;
 using Ringtoets.Common.Data.TestUtil.IllustrationPoints;
+using Ringtoets.Common.Forms.PresentationObjects;
 using Ringtoets.Common.Forms.TestUtil;
 using Ringtoets.Common.Forms.Views;
 
@@ -288,7 +289,7 @@
 
             // Then
             Assert.AreNotEqual(0, selectionChangedCount);
-            Assert.AreSame(generalResult.TopLevelIllustrationPoints.ElementAt(0), view.Selection);
+            Assert.AreSame(generalResult.TopLevelIllustrationPoints.ElementAt(0), ((SelectedTopLevelFaultTreeIllustrationPoint)view.Selection).TopLevelFaultTreeIllustrationPoint);
 
             mocks.VerifyAll();
         }
@@ -351,7 +352,7 @@
 
             // Then
             Assert.AreEqual(1, selectionChangedCount);
-            Assert.AreSame(generalResult.TopLevelIllustrationPoints.ElementAt(1), view.Selection);
+            Assert.AreSame(generalResult.TopLevelIllustrationPoints.ElementAt(1), ((SelectedTopLevelFaultTreeIllustrationPoint) view.Selection).TopLevelFaultTreeIllustrationPoint);
 
             mocks.VerifyAll();
         }