The Sri Lankans lost their last five wickets for just 23 runs as fast bowler Brett Lee and all-rounder Shane Watson picked up three wickets apiece.