Add check for correct solver type