Fixes unit tests for the code that has already been released

This fixes the unit tests for the current release.