oops - fix bug