added image assertion back in