[test] Make Custom test safer