diff --git a/rtl8720dn/netdriver.go b/rtl8720dn/netdriver.go
index 15489ee..1e64f41 100644
--- a/rtl8720dn/netdriver.go
+++ b/rtl8720dn/netdriver.go
@@ -227,7 +227,7 @@ func (r *RTL8720DN) DisconnectSocket() error {
 		fmt.Printf("DisconnectSocket()\r\n")
 	}
 	switch r.connectionType {
-	case ConnectionTypeTCP:
+	case ConnectionTypeTCP, ConnectionTypeUDP:
 		_, err := r.Rpc_lwip_close(r.socket)
 		if err != nil {
 			return err