I've added the same check to oledmp.c. I had to replace the enums with #defines to get this to work.