Fixes the unit tests