Added type args to remove warnings in tests.
Merge pull request #97 from nicholassm/fix_warnings_in_tests

Remove warnings in tests.