Fixed - 100% coverage (hopefully)