Fix bug causing low coverage