fix remove test