Hotfix for wrong class loading

PR voor hotfix