fix tiny bugs