Fix styling on modal