Cleaned up tests.

Need to remove this. There is no class named `Shot`.