Fixed a bug

Connor