Refactored some code as there was an unassigned reference.