Added a small check