Fixed bug in feature