A couple of small adjustments