Added line of code that was accidentally deleted.