1 files changed, 5 insertions, 4 deletions
diff --git a/src/drivers/St7789.cpp b/src/drivers/St7789.cpp
index f47c2f5..6b66313 100644
--- a/src/drivers/St7789.cpp
+++ b/src/drivers/St7789.cpp
@@ -37,8 +37,8 @@ void St7789::WriteData(uint8_t data) {
 }
 
 
-void St7789::WriteSpi(const uint8_t* data, size_t size) {
-    spi.Write(data, size);
+void St7789::WriteSpi(const uint8_t* data, size_t size, size_t repeat) {
+    spi.Write(data, size, repeat);
 }
 
 void St7789::SoftwareReset() {
@@ -143,10 +143,11 @@ void St7789::BeginDrawBuffer(uint16_t x, uint16_t y, uint16_t width, uint16_t he
 }
 
 void St7789::EndDrawBuffer() {
+  spi.Wait();
 }
 
-void St7789::NextDrawBuffer(const uint8_t *data, size_t size) {
-  WriteSpi(data, size);
+void St7789::NextDrawBuffer(const uint8_t *data, size_t size, size_t repeat) {
+  WriteSpi(data, size, repeat);
 }
 
 void St7789::HardwareReset() {