Made another minor tweak