A small fix for the last commit