fix a bug introduced in prev commit