diff --git a/vardict.wdl b/vardict.wdl
index 4d80738a0030b9c3ccd75286463500f378c4aa74..3fcdb225ff1c085f345272979769ffd6344962c1 100644
--- a/vardict.wdl
+++ b/vardict.wdl
@@ -22,12 +22,14 @@ task VarDict {
         Int geneColumn = 4
 
         String? preCommand
+        Int memory = 4
+        Float memoryMultiplier = 2.0
     }
 
     String toolCommand = if defined(installDir)
         then installDir + "/VarDict"
         else if useJavaVersion
-            then "vardict-java" #probably needs memory stuff
+            then "vardict-java -Xmx${memory}"
             else "vardict"
 
     command {
@@ -54,4 +56,8 @@ task VarDict {
     output {
         File vcfFile = outputVcf
     }
+
+    runtime {
+        memory: ceil(memory * memoryMultiplier)
+    }
 }