When you return to the US, continue using your H1B visa to work instead of EAD so that you are still on H1 status. Good luck.