Fixed bug in testing.