minor fix for 1st solution to P0301