fix forget implementation