Fixes check mode in test