Fix for 'forget' function