Fixing bug in previous challenge.