fix lightning bug