TODO: fix remove docs