Minor cleanup, use a single line instead of two