Fix for optional args