fix stupid bug from last version