Fixed a minor bug with previous engine update