Removed some dead code from the test