small fix

thanks to @Rail