small fix to #9 solution