quick fix on bug