Simplify the optional parameter logic