Fixed a small bug... ooops