Fixed an issue with the return value.