Forgot the conditional case in previous fix.