Fix feature test

- Fix feature test by removing class from background data.