Simplify a bit of code

Same code, just a bit simpler.