fixed small error due to cleanup