Fixed source to compile on VS