Remove warnings in tests.

This is a small commit that simply removes warnings that hurt my eyes in an otherwise very clean code base. :-)

- Added type args to remove warnings in tests.