diff --git a/v3io_gputils/__init__.py b/v3io_gputils/__init__.py
index ffcc925..156d6f9 100644
--- a/v3io_gputils/__init__.py
+++ b/v3io_gputils/__init__.py
@@ -1 +1 @@
-__version__ = '0.0.3'
+__version__ = '0.0.4'
diff --git a/v3io_gputils/mpijob.py b/v3io_gputils/mpijob.py
index fa076e5..e5f856e 100644
--- a/v3io_gputils/mpijob.py
+++ b/v3io_gputils/mpijob.py
@@ -6,7 +6,7 @@
 from kubernetes.client.rest import ApiException
 
 _mpijob_template = {
- 'apiVersion': 'kubeflow.org/v1alpha1',
+ 'apiVersion': 'kubeflow.org/v1alpha2',
  'kind': 'MPIJob',
  'metadata': {
      'name': '',
@@ -52,7 +52,7 @@ class MpiJob:
 
     """
     group = 'kubeflow.org'
-    version = 'v1alpha1'
+    version = 'v1alpha2'
     plural = 'mpijobs'
 
     def __init__(self, name, image=None, command=None,