Fixed import and added warning