small changes

maybe a fix?