Fixed another return blocking code

Fixed another return blocking code