Oh damn forgot to updated one place