fixed that one bug