Some fix-ups to viz, including with the scrolling

1 files changed, 2 insertions, 3 deletions

diff --git a/unit_trace/viz/viewer.py b/unit_trace/viz/viewer.py
index ee5c230..120c7e7 100644
--- a/unit_trace/viz/viewer.py
+++ b/unit_trace/viz/viewer.py
@@ -377,12 +377,11 @@ class MainWindow(gtk.Window):
         
     def switch_page(self, widget, page, page_num):
         if self.notebook.get_nth_page(self.notebook.last_page) is not None:
-            hadj = self.notebook.get_nth_page(self.notebook.last_page).get_hadjustment()
-            old_value = hadj.get_value()
+            old_value = self.notebook.get_nth_page(self.notebook.last_page).get_hadjustment().get_value()
             old_ofs = self.notebook.get_nth_page(self.notebook.last_page).get_graph_area().get_graph().get_origin()[0]
             new_ofs = self.notebook.get_nth_page(page_num).get_graph_area().get_graph().get_origin()[0]
             new_value = old_value - old_ofs + new_ofs
-            hadj.set_value(new_value)
+            self.notebook.get_nth_page(page_num).get_hadjustment().set_value(new_value)
         
         self.notebook.last_page = page_num