fix silly bugs.

sorry, extends a wrong class...