Changed back one line in logic.cpp