Fix the missing case in >>=