Cleanup from code review.

Signed-off-by: Alex Wood <abwood@gmail.com>