Code Review

Yeah, I caught this goof in my current branch have fixed this up as well.