Fix unreachable code issue