diff --git a/src/com/mirantis/mk/Tox.groovy b/src/com/mirantis/mk/Tox.groovy
index 43f30fb..22259c6 100644
--- a/src/com/mirantis/mk/Tox.groovy
+++ b/src/com/mirantis/mk/Tox.groovy
@@ -5,13 +5,15 @@
 
 /**
  * Run tox
+ * @param args tox run arguments
  *
- * @param args string with tox arguments
- * @param returnStdout return stdout from tox
  */
 
-def runTox(String args, boolean returnStdout = true){
-    return docker.image('docker-prod-local.docker.mirantis.net/mirantis/external/tox').inside {
+def runTox(String args){
+    image = params['image'] ?: "docker-prod-local.docker.mirantis.net/mirantis/external/tox"
+    args = params['args'] ?: ""
+    returnStdout = params['returnStdout'].toBoolean() ?: true
+    return docker.image(image).inside {
         sh(script: "tox ${args}", returnStdout: returnStdout)
     }
 }
\ No newline at end of file