fix arduino pbdrv

1 files changed, 2 insertions, 0 deletions

diff --git a/lib/pbdrv/drv/arduino/mod.cpp b/lib/pbdrv/drv/arduino/mod.cpp
index 518a8a4..8a38a5b 100644
--- a/lib/pbdrv/drv/arduino/mod.cpp
+++ b/lib/pbdrv/drv/arduino/mod.cpp
@@ -9,6 +9,7 @@
 #include <stdint.h>
 
 #include "../../pb.h"
+#include "../../pb-mod.h"
 #include "mod.h"
 
 static void recv_event(int bytes) {
@@ -31,5 +32,6 @@ __weak void pbdrv_i2c_send(i2c_addr_t addr, const uint8_t * buf, size_t sz) {
 	Wire.beginTransmission((int) addr);
 	Wire.write(buf, sz);
 	Wire.endTransmission(true);
+	Wire.setWireTimeout(PB_TIMEOUT_US, true);
 }