Fixed outdated piece of code.