Fix potential TypeError
0.0.2