diff --git a/tests/TriacOut/test_TriacOut.cpp b/tests/TriacOut/test_TriacOut.cpp
index 3f95998..7e01a43 100644
--- a/tests/TriacOut/test_TriacOut.cpp
+++ b/tests/TriacOut/test_TriacOut.cpp
@@ -42,17 +42,17 @@ TEST(test_TriacOut, TriacOut_SetupPins)
     //Expect that pin PA1 is set to output
     mock().expectOneCall("RegEdit_SetBit")
         .withPointerParameter("reg", (void *) &PORTA.DIR)
-        .withUnsignedIntParameter("bit_num", (1<<1));
+        .withUnsignedIntParameter("bit_num", G1);
 
     //Expect that pin PB3 is set to output
     mock().expectOneCall("RegEdit_SetBit")
         .withPointerParameter("reg", (void *) &PORTB.DIR)
-        .withUnsignedIntParameter("bit_num", (1<<3));
+        .withUnsignedIntParameter("bit_num", G2);
 
     //Expect that pin PB2 is set to output
     mock().expectOneCall("RegEdit_SetBit")
         .withPointerParameter("reg", (void *) &PORTB.DIR)
-        .withUnsignedIntParameter("bit_num", (1<<2));
+        .withUnsignedIntParameter("bit_num", G3);
 
     TriacOut_SetupPins();
 }