fix wrong code