Solved bug, (this class is ugly)