Fixed hard-coded reference
Fixed hard-coded reference