Fix test for remove function, now that pickers are detached