Fixed.

Must have copied some of the coding in wrong. :)