Good catch. I need to add that logic to the else.