I got it back working by adding self.page_addressing = page_addressing in the SSD1306_SPI class