use columns from header

diff --git a/lib/MojoFacets/Code.pm b/lib/MojoFacets/Code.pm
index 983bcc7..8111c36 100644 (file)
--- a/lib/MojoFacets/Code.pm
+++ b/lib/MojoFacets/Code.pm
@@ -15,6 +15,7 @@ sub index {
 
        $self->redirect_to('/data/columns') unless $self->session('columns');
        my $columns = { map { $_ => 1 } @{ $self->session('columns') } };
 
        $self->redirect_to('/data/columns') unless $self->session('columns');
        my $columns = { map { $_ => 1 } @{ $self->session('columns') } };

+       $columns->{$_} = 1 foreach @{ $self->session('header') };

 
        my $dir = $self->_code_dir;
 
 
        my $dir = $self->_code_dir;