Test modified for changed api