Make the fix shorter by @Bigpet