Minor DH flow fix