Fixing problem with feature tests.