fix image generator
Merge pull request #1 from vijayanandnandam/images-generator-fix

fix image generator