Test didn't cover all cases