Fixed a test error