Woops, update the unit tests references too