Fix compile problem.

This is a dirty hack.