Please refactor line 225 to reflect this change