minor fix for solution 1