Adapted test to changed API.