diff --git a/autoPyTorch/__version__.py b/autoPyTorch/__version__.py
index 10b95664f..94b9a71f5 100644
--- a/autoPyTorch/__version__.py
+++ b/autoPyTorch/__version__.py
@@ -1,4 +1,4 @@
 """Version information."""
 
 # The following line *must* be the last in the module, exactly as formatted:
-__version__ = "0.1.1"
+__version__ = "0.2"
diff --git a/setup.py b/setup.py
index 4198ff6b5..7a8ef8918 100755
--- a/setup.py
+++ b/setup.py
@@ -21,7 +21,7 @@
 # noinspection PyInterpreter
 setuptools.setup(
     name="autoPyTorch",
-    version="0.1.1",
+    version="0.2",
     author="AutoML Freiburg",
     author_email="eddiebergmanhs@gmail.com",
     description=("Auto-PyTorch searches neural architectures using smac"),
diff --git a/test/test_pipeline/test_tabular_classification.py b/test/test_pipeline/test_tabular_classification.py
index ab21897b8..c679b931d 100644
--- a/test/test_pipeline/test_tabular_classification.py
+++ b/test/test_pipeline/test_tabular_classification.py
@@ -9,6 +9,8 @@
     UniformIntegerHyperparameter,
 )
 
+from flaky import flaky
+
 import numpy as np
 
 import pytest
@@ -59,6 +61,7 @@ def _assert_pipeline_search_space(self, pipeline, search_space_updates):
             elif isinstance(hyperparameter, CategoricalHyperparameter):
                 assert update.value_range == hyperparameter.choices
 
+    @flaky(max_runs=3)
     def test_pipeline_fit(self, fit_dictionary_tabular, exclude):
         """This test makes sure that the pipeline is able to fit
         given random combinations of hyperparameters across the pipeline"""