follow-up fix to the last commit