summaryrefslogtreecommitdiff
path: root/rust/tests/binary_writer.rs
diff options
context:
space:
mode:
Diffstat (limited to 'rust/tests/binary_writer.rs')
-rw-r--r--rust/tests/binary_writer.rs4
1 files changed, 2 insertions, 2 deletions
diff --git a/rust/tests/binary_writer.rs b/rust/tests/binary_writer.rs
index bc9c2165..632d0e64 100644
--- a/rust/tests/binary_writer.rs
+++ b/rust/tests/binary_writer.rs
@@ -1,5 +1,5 @@
use binaryninja::binary_reader::BinaryReader;
-use binaryninja::binary_view::BinaryViewBase;
+use binaryninja::binary_view::{BinaryViewBase, BinaryViewExt};
use binaryninja::binary_writer::BinaryWriter;
use binaryninja::headless::Session;
use rstest::*;
@@ -17,7 +17,7 @@ fn test_binary_writer_seek(_session: &Session) {
let out_dir = env!("OUT_DIR").parse::<PathBuf>().unwrap();
let view = binaryninja::load(out_dir.join("atox.obj")).expect("Failed to create view");
let mut writer = BinaryWriter::new(&view);
- let end_offset = view.len();
+ let end_offset = view.original_image_base() + view.len();
// Test seeking to a specific position
writer