Fix segfault..