Whoops - fixed a bug