Fixed error in from refactoring functional tests