diff --git a/src/main.zig b/src/main.zig
index 9f7f21e..12fc5e9 100644
--- a/src/main.zig
+++ b/src/main.zig
@@ -120,8 +120,10 @@ pub fn main() !void {
     assert(status == hlo.HAILO_SUCCESS);
     std.debug.print("input flushed!\n", .{});
 
+
+
     var output_frame_size: usize = 0;
-    status = hlo.hailo_get_output_vstream_frame_size(input_vstreams[0], &output_frame_size);
+    status = hlo.hailo_get_output_vstream_frame_size(output_vstreams[0], &output_frame_size);
     assert(status == hlo.HAILO_SUCCESS);
     std.debug.print("output frame size is: {d}\n", .{ output_frame_size });
 
@@ -130,6 +132,7 @@ pub fn main() !void {
 
     status = hlo.hailo_vstream_read_raw_buffer(output_vstreams[0], output_data.ptr, output_frame_size);
     assert(status == hlo.HAILO_SUCCESS);
+    std.debug.print("Output finished reading! Cleanup time\n", .{});
 
 
     _ = hlo.hailo_release_output_vstreams(&output_vstreams, output_vstream_size);