Fix acidental deletion of class