Got rid of extra import