diff --git a/manifests/init.pp b/manifests/init.pp
index d5a1da3396ccb96cdb4952fc61f8edfe06146013..3fe24263b42c9806d241fbb85db36a70c84bf26d 100644
--- a/manifests/init.pp
+++ b/manifests/init.pp
@@ -244,4 +244,13 @@ class aim_control (
     backuprotate   => '7',
     time           => ['1','0'],
   }
+
+  # Ensure the MySQL directory is managed properly
+  file { '/var/lib/mysql':
+    ensure => 'directory',
+    owner  => 'mysql',
+    group  => 'mysql',
+    mode   => '0755',
+    require => Class['mysql::server'],  # Ensure MySQL is configured before setting permissions
+  }
 }