remove function fix