small tweak to functions