replace explicit Just with returns