Slight tweak to prob