Add missing resource cleanup