Minor tweak to source