That will obviously work too (just ran the tests to confirm), furthermore it might also prevent other issues in the future.