Fix tests fail