oops, forgot one small change in the last commit