Fixes a misplaced import