small fix not perfect