Fix missing cast