Fixed compile error I overlooked