fix mistakes from last commit