Fixed an issue with test data