Fix SEGV bug.
Cleanup.