Fixing wrong class call