slight logic bug fix