Fix Missing Return Statement