Increase junit coverage