fix the wrong solution