Fixed broken return statement