Fixed minor bugs in testing file.