@@ -216,10 +216,15 @@
 		@throw [OFInvalidArgumentException
 		    exceptionWithClass: [self class]];
 
 	dataStorage = dataStorage_;
 }
+
+- (XMPPConnection*)connection
+{
+	return connection;
+}
 
 - (id <XMPPStorage>)dataStorage
 {
 	return dataStorage;
 }