Fix wrong classname in tests