fix bug introduced by last commit