Fixed a test bug