Fix bug - missing parameter