BUG: Fix a segfault