think i added the right check