Fix restarts controller test