Minor tweak to the prep function