I like to keep the checks with the statements that drive them.  I suggest moving these to the setup function.