fix solution code