diff --git a/build.gradle b/build.gradle
index 6db69a9..606681c 100644
--- a/build.gradle
+++ b/build.gradle
@@ -48,12 +48,10 @@ tasks.register('smali') {
     doLast {
         mkdir('build/dex')
         javaexec {
-            classpath = fileTree('tools/d2j').filter { it.name.endsWith('.jar') }
+            classpath = file('tools/smali-2.5.2.jar')
             println(classpath.files)
 
-            mainClass = 'com.googlecode.d2j.smali.SmaliCmd'
-
-            args = ['-o', 'build/dex/classes.dex', 'build/smali-merged/smali']
+            args = ['a', 'build/smali-merged/smali', '-o', 'build/dex/classes.dex']
         }
     }
 }