Minor change to let the code run on 1.0.2 (hopefully)