Fixed the small bug - frig stupid missight...