Fixed a small bug in cluster