Cover last few lines with tests