Minor changes to fix shared repo