Fix the logic inside forget() method