diff --git a/apps/etterlatte-behandling/src/test/kotlin/behandling/BehandlingFactoryTest.kt b/apps/etterlatte-behandling/src/test/kotlin/behandling/BehandlingFactoryTest.kt
index c69cce5910d..33875eec7ec 100644
--- a/apps/etterlatte-behandling/src/test/kotlin/behandling/BehandlingFactoryTest.kt
+++ b/apps/etterlatte-behandling/src/test/kotlin/behandling/BehandlingFactoryTest.kt
@@ -52,6 +52,7 @@ import no.nav.etterlatte.libs.common.oppgave.opprettNyOppgaveMedReferanseOgSak
 import no.nav.etterlatte.libs.common.person.AdressebeskyttelseGradering
 import no.nav.etterlatte.libs.common.sak.Sak
 import no.nav.etterlatte.libs.common.sak.SakId
+import no.nav.etterlatte.libs.common.sak.SakMedGraderingOgSkjermet
 import no.nav.etterlatte.libs.common.tidspunkt.Tidspunkt
 import no.nav.etterlatte.libs.common.tidspunkt.toLocalDatetimeNorskTid
 import no.nav.etterlatte.libs.common.tidspunkt.toLocalDatetimeUTC
@@ -236,6 +237,13 @@ internal class BehandlingFactoryTest {
         } returns Unit
         coEvery { grunnlagService.leggInnNyttGrunnlag(any(), any(), any()) } returns Unit
         every { oppgaveService.opprettOppgave(any(), any(), any(), any(), any(), gruppeId = any()) } returns mockOppgave
+        every { sakServiceMock.hentGraderingForSak(any(), any()) } returns
+            SakMedGraderingOgSkjermet(
+                opprettetBehandling.sak.id,
+                AdressebeskyttelseGradering.UGRADERT,
+                false,
+                opprettetBehandling.sak.enhet,
+            )
 
         val resultat =
             behandlingFactory
@@ -280,8 +288,7 @@ internal class BehandlingFactoryTest {
         coVerify(exactly = 1) { grunnlagService.leggInnNyttGrunnlag(any(), any(), any()) }
         verify(exactly = 1) {
             sakServiceMock.oppdaterAdressebeskyttelse(any(), any())
-            sakServiceMock.oppdaterEnhetForSaker(any())
-            sakServiceMock.markerSakerMedSkjerming(any(), any())
+            sakServiceMock.hentGraderingForSak(any(), any())
         }
     }
 
@@ -328,6 +335,13 @@ internal class BehandlingFactoryTest {
                 listOf("Gjenlevende"),
             )
 
+        every { sakServiceMock.hentGraderingForSak(any(), any()) } returns
+            SakMedGraderingOgSkjermet(
+                opprettetBehandling.sak.id,
+                AdressebeskyttelseGradering.UGRADERT,
+                false,
+                opprettetBehandling.sak.enhet,
+            )
         every { sakServiceMock.finnSak(any()) } returns opprettetBehandling.sak
         every { behandlingDaoMock.opprettBehandling(capture(behandlingOpprettes)) } returns Unit
         every { behandlingDaoMock.hentBehandling(capture(behandlingHentes)) } returns opprettetBehandling
@@ -377,8 +391,7 @@ internal class BehandlingFactoryTest {
         coVerify { grunnlagService.leggInnNyttGrunnlag(any(), any(), any()) }
         verify(exactly = 1) {
             sakServiceMock.oppdaterAdressebeskyttelse(any(), any())
-            sakServiceMock.oppdaterEnhetForSaker(any())
-            sakServiceMock.markerSakerMedSkjerming(any(), any())
+            sakServiceMock.hentGraderingForSak(any(), any())
         }
     }
 
@@ -425,6 +438,13 @@ internal class BehandlingFactoryTest {
                 listOf("Gjenlevende"),
             )
 
+        every { sakServiceMock.hentGraderingForSak(any(), any()) } returns
+            SakMedGraderingOgSkjermet(
+                underArbeidBehandling.sak.id,
+                AdressebeskyttelseGradering.UGRADERT,
+                false,
+                underArbeidBehandling.sak.enhet,
+            )
         every { sakServiceMock.finnSak(any()) } returns underArbeidBehandling.sak
         every { behandlingDaoMock.opprettBehandling(capture(behandlingOpprettes)) } returns Unit
         every { behandlingDaoMock.hentBehandling(capture(behandlingHentes)) } returns underArbeidBehandling
@@ -499,8 +519,7 @@ internal class BehandlingFactoryTest {
         }
         verify(exactly = 1) {
             sakServiceMock.oppdaterAdressebeskyttelse(any(), any())
-            sakServiceMock.oppdaterEnhetForSaker(any())
-            sakServiceMock.markerSakerMedSkjerming(any(), any())
+            sakServiceMock.hentGraderingForSak(any(), any())
         }
     }
 
@@ -805,7 +824,13 @@ internal class BehandlingFactoryTest {
                 listOf("Avdoed"),
                 listOf("Gjenlevende"),
             )
-
+        every { sakServiceMock.hentGraderingForSak(any(), any()) } returns
+            SakMedGraderingOgSkjermet(
+                nyBehandling.sak.id,
+                AdressebeskyttelseGradering.UGRADERT,
+                false,
+                nyBehandling.sak.enhet,
+            )
         every { sakServiceMock.finnSak(any()) } returns nyBehandling.sak
         every { behandlingDaoMock.opprettBehandling(capture(behandlingOpprettes)) } returns Unit
         every { behandlingDaoMock.hentBehandling(capture(behandlingHentes)) } returns nyBehandling
@@ -937,8 +962,7 @@ internal class BehandlingFactoryTest {
         }
         verify(exactly = 2) {
             sakServiceMock.oppdaterAdressebeskyttelse(any(), any())
-            sakServiceMock.oppdaterEnhetForSaker(any())
-            sakServiceMock.markerSakerMedSkjerming(any(), any())
+            sakServiceMock.hentGraderingForSak(any(), any())
         }
     }
 
@@ -986,6 +1010,13 @@ internal class BehandlingFactoryTest {
             )
 
         every { sakServiceMock.finnSak(any()) } returns nyBehandling.sak
+        every { sakServiceMock.hentGraderingForSak(any(), any()) } returns
+            SakMedGraderingOgSkjermet(
+                nyBehandling.sak.id,
+                AdressebeskyttelseGradering.UGRADERT,
+                false,
+                nyBehandling.sak.enhet,
+            )
         every { behandlingDaoMock.opprettBehandling(capture(behandlingOpprettes)) } returns Unit
         every { behandlingDaoMock.hentBehandling(capture(behandlingHentes)) } returns nyBehandling
         every {
@@ -1114,9 +1145,8 @@ internal class BehandlingFactoryTest {
             sakServiceMock.finnSak(any())
         }
         verify(exactly = 1) {
+            sakServiceMock.hentGraderingForSak(any(), any())
             sakServiceMock.oppdaterAdressebeskyttelse(any(), any())
-            sakServiceMock.oppdaterEnhetForSaker(any())
-            sakServiceMock.markerSakerMedSkjerming(any(), any())
         }
     }
 
@@ -1164,7 +1194,13 @@ internal class BehandlingFactoryTest {
                 kilde = Vedtaksloesning.GJENNY,
                 sendeBrev = true,
             )
-
+        every { sakServiceMock.hentGraderingForSak(any(), any()) } returns
+            SakMedGraderingOgSkjermet(
+                sak.id,
+                AdressebeskyttelseGradering.UGRADERT,
+                false,
+                sak.enhet,
+            )
         every { sakServiceMock.finnEllerOpprettSakMedGrunnlag(any(), any()) } returns sak
         every { sakServiceMock.finnSak(any<SakId>()) } returns sak
         every { behandlingDaoMock.opprettBehandling(capture(behandlingOpprettes)) } just Runs
@@ -1225,8 +1261,7 @@ internal class BehandlingFactoryTest {
         }
         verify(exactly = 1) {
             sakServiceMock.oppdaterAdressebeskyttelse(any(), any())
-            sakServiceMock.oppdaterEnhetForSaker(any())
-            sakServiceMock.markerSakerMedSkjerming(any(), any())
+            sakServiceMock.hentGraderingForSak(any(), any())
         }
     }