Fixed a small crash