Fix cleanup error