fix forget function