Tidy a wee bit of code.