Moved condition so we don't get an empty div