2 files changed, 4 insertions, 0 deletions
diff --git a/ui/reflectionview.h b/ui/reflectionview.h
index 667a9620..6439fbe4 100644
--- a/ui/reflectionview.h
+++ b/ui/reflectionview.h
@@ -10,6 +10,7 @@
 
 class ContextMenuManager;
 class DisassemblyContainer;
+class FlowGraphWidget;
 class Menu;
 class ViewFrame;
 
@@ -33,6 +34,8 @@ public:
 	ReflectionView(ViewFrame* frame, BinaryViewRef data);
 	~ReflectionView();
 
+	FlowGraphWidget* getFlowGraphWidget();
+
 	void toggleILSync();
 	void toggleLocationSync();
 
diff --git a/ui/viewframe.h b/ui/viewframe.h
index b4d3413e..b7d28fab 100644
--- a/ui/viewframe.h
+++ b/ui/viewframe.h
@@ -241,6 +241,7 @@ public:
 	virtual ~ViewFrame();
 
 	FileContext* getFileContext() const { return m_context; }
+	DockHandler* getDockHandler() const { return m_docks; }
 
 	QString getTabName();
 	QString getShortFileName();