Tiny bit of code tidying up before a release