diff --git a/sys/arm/nvidia/as3722_gpio.c b/sys/arm/nvidia/as3722_gpio.c
--- a/sys/arm/nvidia/as3722_gpio.c
+++ b/sys/arm/nvidia/as3722_gpio.c
@@ -441,7 +441,7 @@
 	}
 	mode = as3722_gpio_get_mode(sc, pin, flags);
 	ctrl &= ~(AS3722_GPIO_MODE_MASK << AS3722_GPIO_MODE_SHIFT);
-	ctrl |= AS3722_MODE_PUSH_PULL << AS3722_GPIO_MODE_SHIFT;
+	ctrl |= mode << AS3722_GPIO_MODE_SHIFT;
 	rv = 0;
 	if (ctrl != sc->gpio_pins[pin]->pin_ctrl_reg) {
 		rv = WR1(sc, AS3722_GPIO0_CONTROL + pin, ctrl);