Another small fix

Fixed something I forgot about