Quick fix a bug