Cleaned up comments in tests