Fix KS Bug

Fix KS Bug