Fixed a crash (I hope :))