fix bug i introduced