Change test class logic to boolean