small fix, had remove instead of update