Remove unnused class
Small fix