现在在做教育回报率的问题，想要检验是否存在自选择的问题

结果方程 log(wage)=a0+a1*highedu+a2*X+e   (1)

                                       1       if highedu*>0

水平方程 highedu=                                                         highedu*=b0+b1*Z+u

                                       0         if highedu*<=0

不知道在stata中用什么命令好，老师说可以用Roy模型，我试过heckman命令，但是那是针对结果方程也存在自选择而言的，马上就要到交稿日期了，却不知道怎么弄才好，求教各位大虾了。