Fixed another oversight