Removed extra wait time and fixed to look for image_id instead of id