diff --git a/layout_styles.module b/layout_styles.module
index 8b09154..480dadf 100644
--- a/layout_styles.module
+++ b/layout_styles.module
@@ -4,3 +4,13 @@
  * @file
  * Layouts Styles module file.
  */
+
+/**
+ * Implements hook_layout_alter().
+ */
+function layout_styles_layout_alter(&$definitions) {
+    $definitions['layout_onecol']->setClass('\Drupal\layout_styles\Plugin\Layout\LayoutBase');
+    $definitions['layout_twocol_section']->setClass('\Drupal\layout_styles\Plugin\Layout\TwoColumnWidth');
+    $definitions['layout_threecol_section']->setClass('\Drupal\layout_styles\Plugin\Layout\ThreeColumnWidth');
+    $definitions['layout_fourcol_section']->setClass('\Drupal\layout_styles\Plugin\Layout\LayoutBase');
+}