Fix solution code