fix bug introduced by previous commits