Fix bug introduced by previous check in