Add flags to tests