1 files changed, 3 insertions, 6 deletions
diff --git a/Swiften/Compress/ZLibCodecompressor.h b/Swiften/Compress/ZLibCodecompressor.h
index 289704d..93babf0 100644
--- a/Swiften/Compress/ZLibCodecompressor.h
+++ b/Swiften/Compress/ZLibCodecompressor.h
@@ -4,12 +4,11 @@
  * See Documentation/Licenses/GPLv3.txt for more information.
  */
 
-#ifndef SWIFTEN_ZLibCodecompressor_H
-#define SWIFTEN_ZLibCodecompressor_H
+#pragma once
 
 #include <zlib.h>
 
-#include "Swiften/Base/ByteArray.h"
+#include <Swiften/Base/SafeByteArray.h>
 
 namespace Swift {
 	class ZLibCodecompressor {
@@ -17,12 +16,10 @@ namespace Swift {
 			ZLibCodecompressor();
 			virtual ~ZLibCodecompressor();
 
-			ByteArray process(const ByteArray& data);
+			SafeByteArray process(const SafeByteArray& data);
 			virtual int processZStream() = 0;
 
 		protected:
 			z_stream stream_;
 	};
 }
-
-#endif