diff --git a/src/main.zig b/src/main.zig
index 6542de6..d860013 100644
--- a/src/main.zig
+++ b/src/main.zig
@@ -7,7 +7,8 @@ const hlo = @cImport({
 const Allocator = std.mem.Allocator;
 const assert = std.debug.assert;
 
-const hef_file = "yolov5s.hef";
+// const hef_file = "yolov5s.hef";
+const hef_file = "./libhailort/examples/hefs/shortcut_net";
 const max_edge_layers = 32;