Would be nicer as a ternary?