Fix tests to use specific version