[CHORE] Assign Ks when calling Out.

Signed-off-by: Amlal El Mahrouss <amlal@nekernel.org>

1 files changed, 3 insertions, 4 deletions

diff --git a/src/kernel/NeKit/Stream.h b/src/kernel/NeKit/Stream.h
index b24038c1..0fac23c1 100644
--- a/src/kernel/NeKit/Stream.h
+++ b/src/kernel/NeKit/Stream.h
@@ -5,6 +5,7 @@
 #ifndef NEKIT_STREAM_H
 #define NEKIT_STREAM_H
 
+#include <CompilerKit/CompilerKit.h>
 #include <NeKit/Config.h>
 #include <NeKit/Ref.h>
 
@@ -14,11 +15,9 @@ template <typename StreamTrait, typename Kind>
 class Stream final {
  public:
   explicit Stream(Ref<Stream> ref) : fStream(ref) {}
-
   ~Stream() = default;
 
-  Stream& operator=(const Stream&) = default;
-  Stream(const Stream&)            = default;
+  NE_COPY_DEFAULT(Stream)
 
   template <typename Data>
   friend Stream<StreamTrait, Kind>& operator>>(Stream<StreamTrait, Kind>& Ks, Ref<Data>& Buf) {
@@ -29,7 +28,7 @@ class Stream final {
   template <typename Data>
   friend Stream<StreamTrait, Kind>& operator<<(Stream<StreamTrait, Kind>& Ks, Ref<Data>& Buf) {
     Ks.fKind = Buf;
-    Ks.fStream->Out(Buf.Leak());
+    *Ks = Ks.fStream->Out(Buf.Leak());
     return *Ks;
   }