fix tiny bug