diff --git a/src/main.cpp b/src/main.cpp
index 2f4eb95..3dacad9 100644
--- a/src/main.cpp
+++ b/src/main.cpp
@@ -168,6 +168,7 @@ void get_eeprom_last_commit(String)
     return;
 }
 
+#ifndef UNIT_TEST // don't define setup() and loop() for unit testing
 void setup()
 {
     #if defined(SUPPORT_EEPROM) && defined(MCU_PICO)
@@ -238,6 +239,7 @@ void loop()
         comPort = (HardwareSerial*) &SECONDARY_SERIAL_PORT;
         bytesAvailable = true;
     }
+    #endif
     if (bytesAvailable)
     {
         handle_command(comPort->readStringUntil('\n'));
@@ -250,4 +252,4 @@ void loop()
     for (int i = 0; i < registered_loop_fn_count; i++)
         registered_loop_functions[i]();
 }
-#endif
\ No newline at end of file
+#endif //ndef UNIT_TEST
\ No newline at end of file