small changes to fix the stupid mistakes