Fix the error -> think more