Small change to accomidate the changes in main project.