Minor cleanup

Yeah, thanks for noticing, I was just about to do a PR :)