added assertion
removed an unused import