fixing stupid type in transit probability... the code is right though