fix bug i just introduced