Fixed issue #1 from the original repo

https://github.com/culturevis/imageplot/issues/1