Update remove test
Fix remove