Fix for missing param (right fix!)