Handle "None" return value from function