Minor imports fix