Small fix

Fixed line I forgot about.