1 files changed, 4 insertions, 2 deletions
diff --git a/lib/frame_stream.h b/lib/frame_stream.h
index bc0b9a2..32fea15 100644
--- a/lib/frame_stream.h
+++ b/lib/frame_stream.h
@@ -26,14 +26,16 @@
  */
 #include "config.h"
 #ifndef __MIAV_FRAME_STREAM_H__
-#define __MIAV_FRAME_STREAM_H__
+#define __MIAV_FRAME_STREAM_H_
+
+#include "frame.h"
 
 class frame_stream {
 public:
   frame_stream() {}
   virtual ~frame_stream() {}
 
-  virtual unsigned char *readFrame() = 0;
+  virtual Frame *readFrame() = 0;
 };