Cleanup
Fix warning