Fixed stupid mistakes