Forgot second piece of fix.