diff --git a/src/test/java/cz/muni/fi/pa165/icehockeymanager/services/TeamServiceImplTest.java b/src/test/java/cz/muni/fi/pa165/icehockeymanager/services/TeamServiceImplTest.java
index 74b7782d4cd7ee7b5cf7d07f5bc767eecda56723..e4d565d2881bbda259df4cdd55ed55a581dfadef 100644
--- a/src/test/java/cz/muni/fi/pa165/icehockeymanager/services/TeamServiceImplTest.java
+++ b/src/test/java/cz/muni/fi/pa165/icehockeymanager/services/TeamServiceImplTest.java
@@ -150,7 +150,7 @@ class TeamServiceImplTest {
         assertThatExceptionOfType(NullPointerException.class).isThrownBy(
                 () -> teamService.addHomeGame(team, null)
         );
-        assertThat(team.getHomeGames()).hasSameElementsAs(List.of());
+        assertThat(team.getHomeGames()).isEmpty();
     }
 
     @Test
@@ -176,7 +176,7 @@ class TeamServiceImplTest {
         assertThatExceptionOfType(NullPointerException.class).isThrownBy(
                 () -> teamService.addAwayGame(team, null)
         );
-        assertThat(team.getAwayGames()).hasSameElementsAs(List.of());
+        assertThat(team.getAwayGames()).isEmpty();
     }
 
     @Test
@@ -188,12 +188,24 @@ class TeamServiceImplTest {
     }
 
     @Test
-    public void removeHomeGame() {
+    public void removeHomeGameOneOfOne() {
         Team team = buildTeam("Florida Panthers");
         Game game = buildGame();
         team.setHomeGames(Set.of(game));
         teamService.removeHomeGame(team, game);
-        assertThat(team.getHomeGames()).hasSameElementsAs(List.of());
+        assertThat(team.getHomeGames()).isEmpty();
+        verify(teamDao).update(team);
+    }
+
+    @Test
+    public void removeHomeGameOnOfMany() {
+        Team team = buildTeam("Florida Panthers");
+        Game game = buildGame();
+        game.setId(2L);
+        Game game2 = buildGame();
+        team.setHomeGames(Set.of(game, game2));
+        teamService.removeHomeGame(team, game);
+        assertThat(team.getHomeGames()).hasSameElementsAs(Set.of(game2));
         verify(teamDao).update(team);
     }