Fixed function after changing import declaration