Minor tweak with api