Clean up code a tiny bit.