Minor styling fix #2