Fixed bug caused by change in image args