1 files changed, 4 insertions, 4 deletions
diff --git a/Swiften/Component/UnitTest/ComponentSessionTest.cpp b/Swiften/Component/UnitTest/ComponentSessionTest.cpp
index 0533645..79d9ab3 100644
--- a/Swiften/Component/UnitTest/ComponentSessionTest.cpp
+++ b/Swiften/Component/UnitTest/ComponentSessionTest.cpp
@@ -1,11 +1,11 @@
 /*
- * Copyright (c) 2010-2013 Remko Tronçon
+ * Copyright (c) 2010-2014 Remko Tronçon
  * Licensed under the GNU General Public License v3.
  * See Documentation/Licenses/GPLv3.txt for more information.
  */
 
 #include <cppunit/extensions/HelperMacros.h>
 #include <cppunit/extensions/TestFactoryRegistry.h>
 #include <deque>
 #include <boost/bind.hpp>
 #include <boost/optional.hpp>
@@ -80,23 +80,23 @@ class ComponentSessionTest : public CppUnit::TestFixture {
 
 		void handleSessionFinished(boost::shared_ptr<Error> error) {
 			sessionFinishedReceived = true;
 			sessionFinishedError = error;
 		}
 
 		class MockSessionStream : public SessionStream {
 			public:
 				struct Event {
-					Event(boost::shared_ptr<Element> element) : element(element), footer(false) {}
+					Event(boost::shared_ptr<ToplevelElement> element) : element(element), footer(false) {}
 					Event(const ProtocolHeader& header) : header(header), footer(false) {}
 					Event() : footer(true) {}
 					
-					boost::shared_ptr<Element> element;
+					boost::shared_ptr<ToplevelElement> element;
 					boost::optional<ProtocolHeader> header;
 					bool footer;
 				};
 
 				MockSessionStream() : available(true), whitespacePingEnabled(false), resetCount(0) {
 				}
 
 				virtual void close() {
 					onClosed(boost::shared_ptr<Error>());
@@ -108,19 +108,19 @@ class ComponentSessionTest : public CppUnit::TestFixture {
 
 				virtual void writeHeader(const ProtocolHeader& header) {
 					receivedEvents.push_back(Event(header));
 				}
 
 				virtual void writeFooter() {
 					receivedEvents.push_back(Event());
 				}
 
-				virtual void writeElement(boost::shared_ptr<Element> element) {
+				virtual void writeElement(boost::shared_ptr<ToplevelElement> element) {
 					receivedEvents.push_back(Event(element));
 				}
 
 				virtual void writeData(const std::string&) {
 				}
 
 				virtual bool supportsTLSEncryption() {
 					return false;
 				}