diff --git a/app/src/main/kotlin/info/appdev/chartexample/MultiLineChartActivity.kt b/app/src/main/kotlin/info/appdev/chartexample/MultiLineChartActivity.kt
index 978b99595..d820d004d 100644
--- a/app/src/main/kotlin/info/appdev/chartexample/MultiLineChartActivity.kt
+++ b/app/src/main/kotlin/info/appdev/chartexample/MultiLineChartActivity.kt
@@ -185,9 +185,7 @@ class MultiLineChartActivity : DemoBase(), OnSeekBarChangeListener, OnChartGestu
             }
 
             R.id.actionToggleStepped -> {
-                val sets = binding.chart1.data!!.dataSets
-
-                for (iSet in sets) {
+                binding.chart1.data?.dataSets?.let { iSet ->
                     val set = iSet as LineDataSet
                     set.lineMode = if (set.lineMode == LineDataSet.Mode.STEPPED)
                         LineDataSet.Mode.LINEAR
@@ -198,9 +196,7 @@ class MultiLineChartActivity : DemoBase(), OnSeekBarChangeListener, OnChartGestu
             }
 
             R.id.actionToggleHorizontalCubic -> {
-                val sets = binding.chart1.data!!.dataSets
-
-                for (iSet in sets) {
+                binding.chart1.data?.dataSets?.let { iSet ->
                     val set = iSet as LineDataSet
                     set.lineMode = if (set.lineMode == LineDataSet.Mode.HORIZONTAL_BEZIER)
                         LineDataSet.Mode.LINEAR