diff --git a/src/libsysprof-ui/sysprof-notebook.c b/src/libsysprof-ui/sysprof-notebook.c
index 30a0aac6..765f4ee6 100644
--- a/src/libsysprof-ui/sysprof-notebook.c
+++ b/src/libsysprof-ui/sysprof-notebook.c
@@ -161,7 +161,7 @@ sysprof_notebook_open (SysprofNotebook *self,
   sysprof_display_open (SYSPROF_DISPLAY (display), file);
 }
 
-static SysprofDisplay *
+SysprofDisplay *
 sysprof_notebook_get_current (SysprofNotebook *self)
 {
   gint page;
diff --git a/src/libsysprof-ui/sysprof-notebook.h b/src/libsysprof-ui/sysprof-notebook.h
index c3f3d3df..0b467bd3 100644
--- a/src/libsysprof-ui/sysprof-notebook.h
+++ b/src/libsysprof-ui/sysprof-notebook.h
@@ -22,6 +22,7 @@
 
 #include <gtk/gtk.h>
 
+#include "sysprof-display.h"
 #include "sysprof-version-macros.h"
 
 G_BEGIN_DECLS
@@ -40,13 +41,15 @@ struct _SysprofNotebookClass
 };
 
 SYSPROF_AVAILABLE_IN_ALL
-GtkWidget *sysprof_notebook_new           (void);
+GtkWidget      *sysprof_notebook_new           (void);
 SYSPROF_AVAILABLE_IN_ALL
-void       sysprof_notebook_close_current (SysprofNotebook *self);
+SysprofDisplay *sysprof_notebook_get_current   (SysprofNotebook *self);
 SYSPROF_AVAILABLE_IN_ALL
-void       sysprof_notebook_open          (SysprofNotebook *self,
-                                           GFile           *file);
+void            sysprof_notebook_close_current (SysprofNotebook *self);
 SYSPROF_AVAILABLE_IN_ALL
-void       sysprof_notebook_save          (SysprofNotebook *self);
+void            sysprof_notebook_open          (SysprofNotebook *self,
+                                                GFile           *file);
+SYSPROF_AVAILABLE_IN_ALL
+void            sysprof_notebook_save          (SysprofNotebook *self);
 
 G_END_DECLS