Fix bug when on restore activity