Ironed out small bug.