Fixes bad import