Fixed a 'bug'

I just added a line