Fix one more regression