fix terrible bug