Fixed error with return value