diff --git a/agml/__init__.py b/agml/__init__.py
index 65b7bd9c..b1d218c9 100644
--- a/agml/__init__.py
+++ b/agml/__init__.py
@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-__version__ = '0.6.1'
+__version__ = '0.6.2'
 __all__ = ['data', 'synthetic', 'backend', 'viz', 'io']
 
 
diff --git a/agml/data/exporters/yolo.py b/agml/data/exporters/yolo.py
index 400c70aa..894212a9 100644
--- a/agml/data/exporters/yolo.py
+++ b/agml/data/exporters/yolo.py
@@ -154,7 +154,7 @@ def export_yolo(dataset, yolo_path=None):
     """.format(yolo_data_path, os.path.join(yolo_path, 'data', 'dataset.yaml'))
     log(information)
 
-    return yolo_data_path
+    return {'dataset_path': yolo_data_path, 'metadata_path': os.path.join(yolo_path, 'data', 'dataset.yaml')}