Fix infinite loop due to WIP code.