Simplify return condition

I think this tends to be more clear? Feel free to ignore this request :D