Small regression fix.