Minor refactor.

Signed-off-by: King Lemming <kinglemming@gmail.com>