Fix small issue in load test