diff --git a/makina-compiler/test/xyz/colinholzman/makina/TransitionTest.kt b/makina-compiler/test/xyz/colinholzman/makina/TransitionTest.kt
index 947a97f..0aecb07 100644
--- a/makina-compiler/test/xyz/colinholzman/makina/TransitionTest.kt
+++ b/makina-compiler/test/xyz/colinholzman/makina/TransitionTest.kt
@@ -19,6 +19,8 @@ internal class TransitionTest {
         assertEquals(listOf(s12, s122), transition.getEntrySet())
         transition = Transition(s111, s21)
         assertEquals(listOf(s2, s21), transition.getEntrySet())
+        transition = Transition(s111, s111)
+        assertEquals(listOf(s111), transition.getEntrySet())
     }
 
     @Test
@@ -27,5 +29,7 @@ internal class TransitionTest {
         assertEquals(listOf(s111, s11), transition.getExitSet())
         transition = Transition(s111, s21)
         assertEquals(listOf(s111, s11, s1), transition.getExitSet())
+        transition = Transition(s111, s111)
+        assertEquals(listOf(s111), transition.getExitSet())
     }
 }
\ No newline at end of file