fixed last bug - yay!