epaper2in9b.py set_rotate fails with epdif undefined

[AndyThirtover]

As it says in the title, epdif is undefined.
I tried this:

def set_rotate(self, rotate):
if (rotate == ROTATE_0):
self.rotate = ROTATE_0
self.width = EPD_WIDTH
self.height = EPD_HEIGHT
elif (rotate == ROTATE_90):
self.rotate = ROTATE_90
self.width = EPD_HEIGHT
self.height = EPD_WIDTH
elif (rotate == ROTATE_180):
self.rotate = ROTATE_180
self.width = EPD_WIDTH
self.height = EPD_HEIGHT
elif (rotate == ROTATE_270):
self.rotate = ROTATE_270
self.width = EPD_HEIGHT
self.height = EPD_WIDTH

But rotation didn't happen :(.

Regards

Andy