Small regression fix