Grove - OLED Display 1.12" scrambled output

I received this module, installed the library, and loaded one of the example sketches. All I get is random white dots and only on the bottom 1/2 of the scren. I am thinking I may have a bad screen or something. Pic below. Thanks!

20141107_205102.jpg

I had something similar when I tried to connect it to the GrovePi and use the default library.
The problem was to do with the initialisation code.

There are two OLED screens, 96x96 and 128x64, each with different chips and initialisation sequence.

I was trying to use the 96x96 code on a 128x64 display.
When I tried using the 128x64 code, it started working as expected.