minor fix to avoid class confusion