Fixed tests for coverage