Restore functionality by removing extraneous return