[test] Small fix to test