Fix bug in tests!