Fix wrong code