Idiot Nik! Adding missing use statement