tweak to the logic which reads a little nicer