fixing 1 last issue