Fixed OneClass bug