Yea this needs to be more robust.