minor tweak to tw model