Minor cleanup.

Use a look instead of duplicate code.