Fixed problem with test data