Hotfix/image tests