Fixed error on test suite