diff --git a/deepdog/__init__.py b/deepdog/__init__.py
index e590cf3..07ca84b 100644
--- a/deepdog/__init__.py
+++ b/deepdog/__init__.py
@@ -1,7 +1,7 @@
 import logging
 from deepdog.meta import __version__
 from deepdog.bayes_run import BayesRun
-
+from deepdog.diagnostic import Diagnostic
 
 def get_version():
 	return __version__
diff --git a/deepdog/diagnostic.py b/deepdog/diagnostic.py
index f4584ab..04297a9 100644
--- a/deepdog/diagnostic.py
+++ b/deepdog/diagnostic.py
@@ -69,7 +69,7 @@ class Diagnostic():
 					row = {
 						"model": name,
 						"index": idx,
-						"bounds_px": bounds,
+						"bounds": bounds,
 						"actual_dipole_moment": self.dipoles.dipoles[0].p,
 						"actual_dipole_position": self.dipoles.dipoles[0].s,
 						"actual_dipole_freq": self.dipoles.dipoles[0].w,