fixed seg fault :)