FIX shape issue when y has only one single class + some missing doc