fix bug in previous commit