fixed wrong code