Fix failing removal test