Fix minor bug introduced by last commit