Fix tests change