Fix vim color

Fix vim color