Fix bug in robust