Fixed seg fault. There's another one