Fix cg bug