Small fix (thanks @HybridDog)