Add replacement back into tool UI