Your lawyer is not right. If you used EAD, your status will be changed to AOS (adjustment of status), and I don't think you can go back.