Fix bug in test file