Fix for wrong class name (whats wrong here?)