1 files changed, 4 insertions, 2 deletions

diff --git a/drivers/net/ethernet/mellanox/mlx4/en_selftest.c b/drivers/net/ethernet/mellanox/mlx4/en_selftest.c
index c11d063473e5..03e5f6ac67e7 100644
--- a/drivers/net/ethernet/mellanox/mlx4/en_selftest.c
+++ b/drivers/net/ethernet/mellanox/mlx4/en_selftest.c
@@ -129,8 +129,10 @@ static int mlx4_en_test_speed(struct mlx4_en_priv *priv)
         if (mlx4_en_QUERY_PORT(priv->mdev, priv->port))
                 return -ENOMEM;
 
-        /* The device currently only supports 10G speed */
-        if (priv->port_state.link_speed != SPEED_10000)
+        /* The device supports 1G, 10G and 40G speeds */
+        if (priv->port_state.link_speed != 1000 &&
+            priv->port_state.link_speed != 10000 &&
+            priv->port_state.link_speed != 40000)
                 return priv->port_state.link_speed;
         return 0;
 }