bug fix of last commit. forgot to change other control class.