Fixed another regression