Forgot one change in the last commit