diff --git a/src/InlineEdit/InlineEdit.php b/src/InlineEdit/InlineEdit.php
index ded87e13..815c3f79 100644
--- a/src/InlineEdit/InlineEdit.php
+++ b/src/InlineEdit/InlineEdit.php
@@ -190,7 +190,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 form-control-sm');
 					}