Use match instead of ==