Fixed import (#251)