What problems?

Yes, you can change it to that. It should work both ways, but you are right, that it will be more correct with checking the token.