Small tweak; less code.