diff --git a/docs/tutorials/stateful-application/cassandra.md b/docs/tutorials/stateful-application/cassandra.md
index 1b729dfd42..f64de6f9e5 100644
--- a/docs/tutorials/stateful-application/cassandra.md
+++ b/docs/tutorials/stateful-application/cassandra.md
@@ -159,7 +159,7 @@ Use `kubectl edit` to modify the size of of a Cassandra StatefulSet.
     # and an empty file will abort the edit. If an error occurs while saving this file will be
     # reopened with the relevant failures.
     #
-    apiVersion: apps/v1beta1
+    apiVersion: apps/v1beta2
     kind: StatefulSet
     metadata:
      creationTimestamp: 2016-08-13T18:40:58Z
diff --git a/docs/tutorials/stateful-application/cassandra/cassandra-statefulset.yaml b/docs/tutorials/stateful-application/cassandra/cassandra-statefulset.yaml
index f283acf657..29bb45ad8c 100644
--- a/docs/tutorials/stateful-application/cassandra/cassandra-statefulset.yaml
+++ b/docs/tutorials/stateful-application/cassandra/cassandra-statefulset.yaml
@@ -1,10 +1,15 @@
-apiVersion: "apps/v1beta1"
+apiVersion: apps/v1beta2
 kind: StatefulSet
 metadata:
   name: cassandra
+  labels:
+    app: cassandra
 spec:
   serviceName: cassandra
   replicas: 3
+  selector:
+    matchLabels:
+      app: cassandra
   template:
     metadata:
       labels: