Fixes className for the container