Use correct stage config