Revert "Fix test by passing graph"

This reverts commit 47fb0d6094f3a10e636abe092b4e3b441aaae2b8.
Store DIL in image_saved