Good point, should add the check in there.