Fixing import that was changed by mistake