fixing tests for latest release