Small param fix. Needs rethinking though...