Fix forgotten if!