Fix compile error.
Fix segfaults