Fixed bug in test file