fixed the remove container bug