Fix NoneType error