1 files changed, 9 insertions, 4 deletions

diff --git a/drivers/net/ehea/ehea_ethtool.c b/drivers/net/ehea/ehea_ethtool.c
index 6498455bf97c..679f40ee9572 100644
--- a/drivers/net/ehea/ehea_ethtool.c
+++ b/drivers/net/ehea/ehea_ethtool.c
@@ -196,9 +196,14 @@ static void ehea_get_strings(struct net_device *dev, u32 stringset, u8 *data)
         }
 }
 
-static int ehea_get_stats_count(struct net_device *dev)
+static int ehea_get_sset_count(struct net_device *dev, int sset)
 {
-        return ARRAY_SIZE(ehea_ethtool_stats_keys);
+        switch (sset) {
+        case ETH_SS_STATS:
+                return ARRAY_SIZE(ehea_ethtool_stats_keys);
+        default:
+                return -EOPNOTSUPP;
+        }
 }
 
 static void ehea_get_ethtool_stats(struct net_device *dev,
@@ -207,7 +212,7 @@ static void ehea_get_ethtool_stats(struct net_device *dev,
         int i, k, tmp;
         struct ehea_port *port = netdev_priv(dev);
 
-        for (i = 0; i < ehea_get_stats_count(dev); i++)
+        for (i = 0; i < ehea_get_sset_count(dev, ETH_SS_STATS); i++)
                 data[i] = 0;
         i = 0;
 
@@ -264,7 +269,7 @@ const struct ethtool_ops ehea_ethtool_ops = {
         .get_link = ethtool_op_get_link,
         .set_tso = ethtool_op_set_tso,
         .get_strings = ehea_get_strings,
-        .get_stats_count = ehea_get_stats_count,
+        .get_sset_count = ehea_get_sset_count,
         .get_ethtool_stats = ehea_get_ethtool_stats,
         .get_rx_csum = ehea_get_rx_csum,
         .set_settings = ehea_set_settings,