Fixed a test issue