Address warnings in tests

As of now, three warnings are raised when running the tests.

These should be addressed.