Future Exploration missions will require an Oxygen Generation Assembly (OGA) to electrolyze water to supply oxygen for crew metabolic consumption. The system design will be based on the International Space Station (ISS) OGA but with added improvements based on lessons learned during ISS operations. These improvements will reduce system weight, crew maintenance time and resupply mass from Earth while increasing reliability. Currently, the design team is investigating the feasibility of the upgrades by performing ground tests and analyses. Upgrades being considered include: redesign of the electrolysis cell stack, deletion of the hydrogen dome, replacement of the hydrogen sensors, deletion of the wastewater interface, redesign of the recirculation loop deionizing bed and redesign of the cell stack Power Supply Module. The upgrades will be first demonstrated on the ISS OGA.