Fix to work with atom-shell