Fix wrong code.
Fix wrong code properly this time.