Fix compilation warning

Simple fix to some test cases to remove warnings.