diff --git a/modules/ve-math/ve.ui.MWLatexDialog.js b/modules/ve-math/ve.ui.MWLatexDialog.js
index 402976d..5434106 100644
--- a/modules/ve-math/ve.ui.MWLatexDialog.js
+++ b/modules/ve-math/ve.ui.MWLatexDialog.js
@@ -254,7 +254,7 @@ ve.ui.MWLatexDialog.prototype.updateMwData = function ( mwData ) {
 	ve.ui.MWLatexDialog.super.prototype.updateMwData.call( this, mwData );
 
 	// Get data from dialog
-	display = this.displaySelect.getSelectedItem().getData();
+	display = this.displaySelect.findSelectedItem().getData();
 	id = this.idInput.getValue();
 
 	// Update attributes
diff --git a/modules/ve-math/ve.ui.MWLatexInspector.js b/modules/ve-math/ve.ui.MWLatexInspector.js
index b343b34..53af4a3 100644
--- a/modules/ve-math/ve.ui.MWLatexInspector.js
+++ b/modules/ve-math/ve.ui.MWLatexInspector.js
@@ -116,7 +116,7 @@ ve.ui.MWLatexInspector.prototype.updateMwData = function ( mwData ) {
 	// Parent method
 	ve.ui.MWLatexInspector.super.prototype.updateMwData.call( this, mwData );
 
-	display = this.displaySelect.getSelectedItem().getData();
+	display = this.displaySelect.findSelectedItem().getData();
 	id = this.idInput.getValue();
 
 	mwData.attrs.display = display !== 'default' ? display : undefined;