Little, tiny refactor.

Just a small change.