Sorry for the flaw in coding