Added missin return statement back in.