forgot to change that one function