diff --git a/src/main/java/net/knarcraft/knarlib/util/ConfigHelper.java b/src/main/java/net/knarcraft/knarlib/util/ConfigHelper.java
index c458c0a..64c3bff 100644
--- a/src/main/java/net/knarcraft/knarlib/util/ConfigHelper.java
+++ b/src/main/java/net/knarcraft/knarlib/util/ConfigHelper.java
@@ -1,6 +1,7 @@
 package net.knarcraft.knarlib.util;
 
 import net.knarcraft.knarlib.config.StargateYamlConfiguration;
+import net.knarcraft.knarlib.plugin.ConfigCommentPlugin;
 import net.knarcraft.knarlib.property.ColorConversion;
 import org.bukkit.configuration.ConfigurationSection;
 import org.bukkit.configuration.MemorySection;
@@ -61,15 +62,12 @@ public final class ConfigHelper {
      * @param plugin <p>The plugin to migrate the configuration for</p>
      * @return <p>True if the migration succeeded without any issues</p>
      */
-    public static boolean migrateConfig(@NotNull Plugin plugin) {
+    public static boolean migrateConfig(@NotNull ConfigCommentPlugin plugin) {
         File dataFolder = plugin.getDataFolder();
         File config = new File(dataFolder, CONFIG_FILE);
 
         // Load the old configuration as a normal configuration to prevent old comments from being retained
-        FileConfiguration oldConfiguration = StargateYamlConfiguration.loadConfiguration(plugin, false);
-        if (oldConfiguration == null) {
-            return false;
-        }
+        FileConfiguration oldConfiguration = plugin.getConfig();
 
         //Save the old config just in case something goes wrong
         try {