diff --git a/adafruit_ssd1306.py b/adafruit_ssd1306.py index c354d54..134a665 100644 --- a/adafruit_ssd1306.py +++ b/adafruit_ssd1306.py @@ -255,7 +255,8 @@ def write_framebuf(self): self.pagebuffer[1:] = self.buffer[ 1 + self.width * page : 1 + self.width * (page + 1) ] - self.i2c_device.write(self.pagebuffer) + with self.i2c_device: + self.i2c_device.write(self.pagebuffer) else: with self.i2c_device: self.i2c_device.write(self.buffer)