Fix compiler warning, method must return reference