Slight modification fixing segfault