fixing addClass assumption