Refactor code

Remove redundant reference