Remove warnings/errors in tests. Small fix.