回国后卖掉公司股票的报税问题

之前在美国工作时有stock purchase plan, 所以每次发工资都会按照15%的折扣买公司股票
8月底回国换了份工作，打算明年1月或者2月把公司股票账户里的股票卖掉（公司规定离职后半年内必须卖掉股票）
这样就算2021年的收入了

我没有绿卡，离开美国后原来的工作签证也就作废了
所以最理想的情况，也就只会在2021年作为游客去美国玩几天

现在的问题是，2021年的收入在之后做报税时，我的这笔股票收入是应该按照居住在外国的外国人来，填写W8-Ben表吗？
这样是不是就不用交capital gain tax了？
我这种情况，不需要按照non-resident来报税吧？
谢谢~

关注。字数字数