Added some missing return instructions.