Fixed cleanup.lua