fix seg fault
fix second seg fault