Fix bug in test spec