Adding some additional checks to unions