Fixed missing feature bug