Fixed possible compile error