Fixed bug in return statement check.