Fixed a small error when running all tests