Account for changes in imago.

Signed-off-by: darkestkhan <darkestkhan@gmail.com>