Added a small condition