Fix error @foobarbecue pointed out