Minor regression fix