Just fixing a little segfault caused by the last commit