Update for gettiing CI working