add if statment to make sure things don't break