Fixed one other bug