Fixed wrong dict use