fix minor logic
fix minor logic
Update