Whoops. Fix some logic.