Fixing missing parameter

Fixing missing parameter