Removed unused code in tests