little change
fix memory leak