Fixed class name in test