Fixing a bug with the solution.