Fix small check