Small logic error...