1 files changed, 11 insertions, 1 deletions
diff --git a/src/op_mode/ping.py b/src/op_mode/ping.py
index 610e63cb3..f1d87a118 100755
--- a/src/op_mode/ping.py
+++ b/src/op_mode/ping.py
@@ -18,7 +18,7 @@ import os
 import sys
 import socket
 import ipaddress
-from vyos.util import get_all_vrfs
+from vyos.utils.network import get_all_vrfs
 from vyos.ifconfig import Section
 
 
@@ -90,6 +90,16 @@ options = {
         'type': '<seconds>',
         'help': 'Number of seconds to wait between requests'
     },
+    'ipv4': {
+        'ping': '{command} -4',
+        'type': 'noarg',
+        'help': 'Use IPv4 only'
+    },
+    'ipv6': {
+        'ping': '{command} -6',
+        'type': 'noarg',
+        'help': 'Use IPv6 only'
+    },
     'mark': {
         'ping': '{command} -m {value}',
         'type': '<fwmark>',