dataset removal fix