diff --git a/pyproject.toml b/pyproject.toml
index c3aa5c9..c7a0eb3 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "gen_surv"
-version = "0.6.10"
+version = "0.7.0"
 description = "A Python package for simulating survival data, inspired by the R package genSurv"
 authors = ["Diogo Ribeiro <diogo_dj@hotmail.com>"]
 license = "MIT"
diff --git a/tasks.py b/tasks.py
index a3d64b1..c1ae9d7 100644
--- a/tasks.py
+++ b/tasks.py
@@ -1,4 +1,5 @@
 from invoke.tasks import task
+import shlex
 
 @task
 def test(c):