Fixes incorrect import