Fixing user test scenario with updated API