Remove errors in test code for clean compilation. TODO: Rewrite tests for new functionality of Assignment2.