Fix seg fault...