Can you add unit tests for the methods where you are adding the checks