Oops - small bug but corrected