Fixed the other major logic error