small change to match controller