fix removing for case og for case of one element