Accidentally missed out an exception message at the end of the
file. Added this back in, works now.