Simplify the code a tad.