Not a day without a commit!

Fixed a compilation error + tiny tidy up.