Fix problem with the unit tests script