Added minor tweak