fixed bug, momnets like this make me think I commit too much but meh