Added more assertion for remove test.