Fixed import warning