Fixed an image issue
Merge pull request #7 from slifty/submit_augment

Fixed an image issue