small fix so variable selection could work