Change type to prevent class cast exception (#171)

Change type to prevent class cast exception (#171) #1252