Bugfix for an exception that was happening during features extraction