Fix infinite display