Removed TestCase class
Fixed test