diff --git a/build.gradle b/build.gradle
index bb3585c..aa5c762 100644
--- a/build.gradle
+++ b/build.gradle
@@ -23,7 +23,7 @@ apply plugin: 'kotlin'
 
 dependencies {
     compile "com.google.guava:guava:18.0"
-    compile "com.beust:kobalt:0.170"
+    compile "com.beust:kobalt:0.171"
 }
 sourceSets {
     main.java.srcDirs += 'src/main/kotlin'
diff --git a/kobalt-line-count.iml b/kobalt-line-count.iml
index 5e19edd..c240327 100644
--- a/kobalt-line-count.iml
+++ b/kobalt-line-count.iml
@@ -36,6 +36,6 @@
     <orderEntry type="library" name="Gradle: javax.inject:javax.inject:1" level="project" />
     <orderEntry type="library" name="Gradle: aopalliance:aopalliance:1.0" level="project" />
     <orderEntry type="library" name="Gradle: org.codehaus.plexus:plexus-utils:3.0.20" level="project" />
-    <orderEntry type="library" name="Gradle: com.beust:kobalt:0.170" level="project" />
+    <orderEntry type="library" name="Gradle: com.beust:kobalt:0.171" level="project" />
   </component>
 </module>
\ No newline at end of file
diff --git a/kobalt/wrapper/kobalt-wrapper.properties b/kobalt/wrapper/kobalt-wrapper.properties
index c983ada..46a4966 100644
--- a/kobalt/wrapper/kobalt-wrapper.properties
+++ b/kobalt/wrapper/kobalt-wrapper.properties
@@ -1 +1 @@
-kobalt.version=0.170
\ No newline at end of file
+kobalt.version=0.171
\ No newline at end of file
diff --git a/src/main/kotlin/com/beust/kobalt/plugin/linecount/LineCountMain.kt b/src/main/kotlin/com/beust/kobalt/plugin/linecount/LineCountPlugin.kt
similarity index 94%
rename from src/main/kotlin/com/beust/kobalt/plugin/linecount/LineCountMain.kt
rename to src/main/kotlin/com/beust/kobalt/plugin/linecount/LineCountPlugin.kt
index 330d8f0..df20ec6 100644
--- a/src/main/kotlin/com/beust/kobalt/plugin/linecount/LineCountMain.kt
+++ b/src/main/kotlin/com/beust/kobalt/plugin/linecount/LineCountPlugin.kt
@@ -15,7 +15,7 @@ fun main(argv: Array<String>) {
     com.beust.kobalt.main(argv)
 }
 
-public class LineCountMain : BasePlugin(), KobaltLogger {
+public class LineCountPlugin : BasePlugin(), KobaltLogger {
     companion object {
         const val NAME : String = "kobalt-line-count"
     }
@@ -64,7 +64,7 @@ data class LineCountInfo(var suffix: String = "**kt")
 public fun lineCount(init: LineCountInfo.() -> Unit): LineCountInfo {
     with(LineCountInfo()) {
         init()
-        (Plugins.getPlugin(LineCountMain.NAME) as LineCountMain).info = this
+        (Kobalt.findPlugin(LineCountPlugin.NAME) as LineCountPlugin).info = this
         return this
     }
 }