Fix deprecation warning
0.4.1