fix things from last commit