Minor Tweaks

Fixed a small bug