Update working version of kth smallest tests