Added an extra line of code