Added a return in custom UMIL function