Fix deprecation :boom: