diff --git a/net/sched/cls_flower.c b/net/sched/cls_flower.c
index 1319986693fc..d3997f00c643 100644
--- a/net/sched/cls_flower.c
+++ b/net/sched/cls_flower.c
@@ -1308,7 +1308,7 @@ static int fl_set_enc_opt(struct nlattr **tb, struct fl_flow_key *key,
 				return -EINVAL;
 			}
 
-			if (msk_depth)
+			if (nla_ok(nla_opt_msk, msk_depth))
 				nla_opt_msk = nla_next(nla_opt_msk, &msk_depth);
 			break;
 		case TCA_FLOWER_KEY_ENC_OPTS_VXLAN:
@@ -1341,7 +1341,7 @@ static int fl_set_enc_opt(struct nlattr **tb, struct fl_flow_key *key,
 				return -EINVAL;
 			}
 
-			if (msk_depth)
+			if (nla_ok(nla_opt_msk, msk_depth))
 				nla_opt_msk = nla_next(nla_opt_msk, &msk_depth);
 			break;
 		case TCA_FLOWER_KEY_ENC_OPTS_ERSPAN:
@@ -1374,7 +1374,7 @@ static int fl_set_enc_opt(struct nlattr **tb, struct fl_flow_key *key,
 				return -EINVAL;
 			}
 
-			if (msk_depth)
+			if (nla_ok(nla_opt_msk, msk_depth))
 				nla_opt_msk = nla_next(nla_opt_msk, &msk_depth);
 			break;
 		default: