minor fix for remove in Transform