Correcting refactored reference