Improve test on image