Good point. I'll make a function for generating load_tests function.