Fix small cluster generation