Fix potential error