Fix minor bug introduced in last commit