forgot to add this for the previous fix