quick fix bug