fix lack of imports