Fixed a bug in testing.