I have worked for company A as H-1B and it applied for GC for me. My I-140 (NIW) was approved and I-485 is pending (concurrent filing). Due to project reason, I need to be paid by 50% from company A and another 50% from company B. This is a collobrative project shared by both comapny A and B. So my job duty and salary remains the same as that when the H-1B and I-140 was filed. And I will be physically located at the company A. My lawyer from company A said that I can continue using H-1B for the payroll from company A and using EAD to get paid from company B. I am wondering if it is OK to keep H-1B for company A while work for company B using EAD. Or should I switch to use EAD for both companies? Thanks in advance.