Fix two references, oops