Can we move this to the assert phase?