Fix build

Doesn't segfault, wohoooo!