Add another check, make it even more ugly :)