Fixed a return statement