Add support for upcoming change in Celluloid

Re: celluloid/celluloid#55, this change can be merged once celluloid/celluloid#187 is merged. 

Is there a nicer way to implement this?