diff --git a/src/tests/basictest.cpp b/src/tests/basictest.cpp
index f1df82886049386e47f0ba2619e147622e8c8ad7..1931953bc74a8efe3403579144341ad626f898f4 100644
--- a/src/tests/basictest.cpp
+++ b/src/tests/basictest.cpp
@@ -150,8 +150,11 @@ void BasicTest::dumpReceived() {
 	std::cout << "Stream2:\n";
 	std::cout << receivedData2 << "\n";
 }
-
+#if HAVE_SWIFTEN_3
+void BasicTest::handleElement(boost::shared_ptr<Swift::ToplevelElement> element) {
+#else
 void BasicTest::handleElement(boost::shared_ptr<Swift::Element> element) {
+#endif
 	if (stream1_active) {
 		received.push_back(element);
 	}