fixed little regressions