Add correct icon