diff --git a/src/main.zig b/src/main.zig
index 20ca23a..6f7f7a3 100644
--- a/src/main.zig
+++ b/src/main.zig
@@ -26,9 +26,9 @@ pub fn main() void {
     var status: hlo.hailo_status = undefined;
     var vdevice      : hlo.hailo_vdevice = undefined;
     var hef          : hlo.hailo_hef = undefined;
-    var device_params: hlo.hailo_vdevice_params_t = undefined; // std.mem.zeroInit(hlo.hailo_vdevice_params_t, .{});
-    var network_group: hlo.hailo_configured_network_group = null;
-    const network_group_size     : usize = 1;
+    var config_params: hlo.hailo_configure_params_t = undefined; // std.mem.zeroInit(hlo.hailo_vdevice_params_t, .{});
+    var network_group: hlo.hailo_configured_network_group = undefined;
+    var network_group_size     : usize = 1;
     const input_vstream_params   : [max_edge_layers]hlo.hailo_input_vstream_params_by_name_t = undefined;
     const output_vstream_params  : [max_edge_layers]hlo.hailo_output_vstream_params_by_name_t = undefined;
     const output_vstreams        : [max_edge_layers]hlo.hailo_output_vstream = undefined;
@@ -36,10 +36,11 @@ pub fn main() void {
     var input_vstream_size       : usize = max_edge_layers;
     var output_vstream_size      : usize = max_edge_layers;
 
+    // const net_ptr: [*c]usize = &network_group_size; 
+
     // @memset(&input_vstream_params, 0);
     // @memset(&output_vstream_params, 0);
 
-
     status = hlo.hailo_create_vdevice(null, &vdevice);
     assert(status == hlo.HAILO_SUCCESS);
 
@@ -48,12 +49,19 @@ pub fn main() void {
 
     std.debug.print("vdevice inited, hef file created!\n", .{});
     
-    status = hlo.hailo_init_configure_params(hef, hlo.HAILO_STREAM_INTERFACE_PCIE, @ptrCast(&device_params));
+    status = hlo.hailo_init_configure_params_by_vdevice(hef, vdevice, &config_params);
     assert(status == hlo.HAILO_SUCCESS);
 
+    std.debug.print("Size of network_group: {any}\n", .{ @sizeOf(hlo.hailo_configured_network_group) });
     std.debug.print("Init configure params complete\n", .{});
 
-    status = hlo.hailo_configure_vdevice(vdevice, hef, @ptrCast(&device_params), &network_group, @constCast(&network_group_size));
+    assert(vdevice != null);
+    assert(hef != null);
+    // assert(&config_params != null);
+    // assert(network_group != null); // configure inits this var
+    assert(network_group_size != 0);
+
+    status = hlo.hailo_configure_vdevice(vdevice, hef, &config_params, &network_group, &network_group_size);
     assert(status == hlo.HAILO_SUCCESS);
 
     std.debug.print("Configure vdevice complete!\n", .{});