diff --git a/graphs/__init__.py b/graphs/__init__.py
index e69de29bb..1f4e13661 100644
--- a/graphs/__init__.py
+++ b/graphs/__init__.py
@@ -0,0 +1,22 @@
+# =============================================================================
+# Copyright (C) 2010 Diego Duclos
+#
+# This file is part of pyfa.
+#
+# pyfa is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# pyfa is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with pyfa. If not, see .
+# =============================================================================
+
+
+from .gui.canvasPanel import graphFrame_enabled
+from .gui.frame import GraphFrame
diff --git a/graphs/gui/__init__.py b/graphs/gui/__init__.py
index 10c552c17..ea1a4f707 100644
--- a/graphs/gui/__init__.py
+++ b/graphs/gui/__init__.py
@@ -16,6 +16,3 @@
# You should have received a copy of the GNU General Public License
# along with pyfa. If not, see .
# =============================================================================
-
-
-from .frame import GraphFrame
diff --git a/gui/mainFrame.py b/gui/mainFrame.py
index b82c8337a..9c126bb54 100644
--- a/gui/mainFrame.py
+++ b/gui/mainFrame.py
@@ -37,7 +37,7 @@ import gui.fitCommands as cmd
import gui.globalEvents as GE
from eos.config import gamedata_date, gamedata_version
from eos.modifiedAttributeDict import ModifiedAttributeDict
-from graphs.gui import GraphFrame
+from graphs import GraphFrame
from gui.additionsPane import AdditionsPane
from gui.bitmap_loader import BitmapLoader
from gui.builtinMarketBrowser.events import ItemSelected
diff --git a/gui/mainMenuBar.py b/gui/mainMenuBar.py
index 91d383bc3..b7f37075a 100644
--- a/gui/mainMenuBar.py
+++ b/gui/mainMenuBar.py
@@ -21,9 +21,9 @@
import wx
import config
+import graphs
from service.character import Character
from service.fit import Fit
-from graphs.gui import canvasPanel as graphCanvasPanel
import gui.globalEvents as GE
from gui.bitmap_loader import BitmapLoader
@@ -101,7 +101,7 @@ class MainMenuBar(wx.MenuBar):
graphFrameItem = wx.MenuItem(fitMenu, self.graphFrameId, "&Graphs\tCTRL+G")
graphFrameItem.SetBitmap(BitmapLoader.getBitmap("graphs_small", "gui"))
fitMenu.Append(graphFrameItem)
- if not graphCanvasPanel.graphFrame_enabled:
+ if not graphs.graphFrame_enabled:
self.Enable(self.graphFrameId, False)
self.ignoreRestrictionItem = fitMenu.Append(self.toggleIgnoreRestrictionID, "Disable Fitting Re&strictions")