fix remove function