Fix warnings from test output - #6