diff --git a/src/InlineEdit/InlineEdit.php b/src/InlineEdit/InlineEdit.php
index 33733ff7c..167e49eed 100644
--- a/src/InlineEdit/InlineEdit.php
+++ b/src/InlineEdit/InlineEdit.php
@@ -219,7 +219,7 @@ public function addControlsClasses(Container $container): void
 
 					break;
 				default:
-					if ($control->getControl()->getAttribute('class') === null) {
+					if ($control->getControlPrototype()->getAttribute('class') === null) {
 						$control->setAttribute('class', 'form-control input-sm form-control-sm');
 					}